ride.feature_visualisation

Module Contents

Classes

FeatureVisualisable

Adds feature visualisation capabilities to model

Functions

scatter_plot(features[, labels, classes])

Attributes

logger

ride.feature_visualisation.logger

class ride.feature_visualisation.FeatureVisualisable(hparams, *args, **kwargs)

Bases: ride.feature_extraction.FeatureExtractable

Adds feature visualisation capabilities to model

hparams: Ellipsis

static configs() → ride.core.Configs

validate_attributes()

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

ride.feature_visualisation.scatter_plot(features: numpy.array, labels: numpy.array = None, classes: List[str] = None)