ltbio.ml.supervised.models#

Overview#

Submodules#

Classes#

SkLearnModel

A generic machine learning supervised model.

TorchModel

A generic machine learning supervised model.

Contents#

class ltbio.ml.supervised.models.SkLearnModel(design, name: str = None)#

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

digraph inheritance6c8e763b53 { bgcolor=transparent; rankdir=LR; size="8.0, 12.0"; "ABC" [fillcolor=white,fontname="Vera Sans, DejaVu Sans, Liberation Sans, Arial, Helvetica, sans",fontsize=10,height=0.25,shape=box,style="setlinewidth(0.5),filled",tooltip="Helper class that provides a standard way to create an ABC using"]; "SkLearnModel" [URL="SkLearnModel/index.html#ltbio.ml.supervised.models.SkLearnModel.SkLearnModel",fillcolor=white,fontname="Vera Sans, DejaVu Sans, Liberation Sans, Arial, Helvetica, sans",fontsize=10,height=0.25,shape=box,style="setlinewidth(0.5),filled",target="_top"]; "SupervisedModel" -> "SkLearnModel" [arrowsize=0.5,style="setlinewidth(0.5)"]; "SupervisedModel" [URL="SupervisedModel/index.html#ltbio.ml.supervised.models.SupervisedModel.SupervisedModel",fillcolor=white,fontname="Vera Sans, DejaVu Sans, Liberation Sans, Arial, Helvetica, sans",fontsize=10,height=0.25,shape=box,style="setlinewidth(0.5),filled",target="_top",tooltip="A generic machine learning supervised model."]; "ABC" -> "SupervisedModel" [arrowsize=0.5,style="setlinewidth(0.5)"]; }

A generic machine learning supervised model.

property non_trainable_parameters#
property trained_parameters#
test(dataset, evaluation_metrics=None, version=None)#
train(dataset, conditions)#
class ltbio.ml.supervised.models.TorchModel(design: torch.nn.Module, name: str = None)#

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

digraph inheritancea6078a7efd { bgcolor=transparent; rankdir=LR; size="8.0, 12.0"; "ABC" [fillcolor=white,fontname="Vera Sans, DejaVu Sans, Liberation Sans, Arial, Helvetica, sans",fontsize=10,height=0.25,shape=box,style="setlinewidth(0.5),filled",tooltip="Helper class that provides a standard way to create an ABC using"]; "SupervisedModel" [URL="SupervisedModel/index.html#ltbio.ml.supervised.models.SupervisedModel.SupervisedModel",fillcolor=white,fontname="Vera Sans, DejaVu Sans, Liberation Sans, Arial, Helvetica, sans",fontsize=10,height=0.25,shape=box,style="setlinewidth(0.5),filled",target="_top",tooltip="A generic machine learning supervised model."]; "ABC" -> "SupervisedModel" [arrowsize=0.5,style="setlinewidth(0.5)"]; "TorchModel" [URL="TorchModel/index.html#ltbio.ml.supervised.models.TorchModel.TorchModel",fillcolor=white,fontname="Vera Sans, DejaVu Sans, Liberation Sans, Arial, Helvetica, sans",fontsize=10,height=0.25,shape=box,style="setlinewidth(0.5),filled",target="_top"]; "SupervisedModel" -> "TorchModel" [arrowsize=0.5,style="setlinewidth(0.5)"]; }

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)#