ride.feature_extraction

Module Contents

Classes

FeatureExtractable

Adds feature extraction capabilities to model

Attributes

logger

ride.feature_extraction.logger

class ride.feature_extraction.FeatureExtractable(hparams: pytorch_lightning.utilities.parsing.AttributeDict, *args, **kwargs)

Bases: ride.core.RideMixin

Adds feature extraction capabilities to model

hparams: Ellipsis

static configs() → ride.core.Configs

validate_attributes()

on_init_end(hparams, *args, **kwargs)

metrics_epoch(preds: torch.Tensor, targets: torch.Tensor, prefix: str = None, clear_extracted_features=True, *args, **kwargs) → ride.metrics.MetricDict