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.RideMixinAdds 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]¶