ride.feature_extraction
¶
Module Contents¶
Classes¶
Adds feature extraction capabilities to model |
Attributes¶
- class ride.feature_extraction.FeatureExtractable(hparams: pytorch_lightning.utilities.parsing.AttributeDict, *args, **kwargs)[source]¶
Bases:
ride.core.RideMixin
Adds feature extraction capabilities to model
- static configs() ride.core.Configs [source]¶
- metrics_epoch(preds: torch.Tensor, targets: torch.Tensor, prefix: str = None, clear_extracted_features=True, *args, **kwargs) ride.metrics.MetricDict [source]¶