`ltbio.ml.supervised.models.TorchModel`#

Overview#

Classes#

A generic machine learning supervised model.

Contents#

class ltbio.ml.supervised.models.TorchModel.TorchModel(design: torch.nn.Module, name: str = None)#

Bases: ltbio.ml.supervised.models.SupervisedModel.SupervisedModel

A generic machine learning supervised model.

property non_trainable_parameters#

property trained_parameters#

DEVICE#

save_design(path: str)#

shapes_summary(dataset: ltbio.ml.datasets.BiosignalDataset.BiosignalDataset)#

test(dataset, evaluation_metrics=(), version=None)#

train(dataset, conditions, n_subprocesses: int = 0, track_memory: bool = False)#