scvi.train.ClassifierTrainingPlanConfig

class scvi.train.ClassifierTrainingPlanConfig(lr=0.001, weight_decay=1e-06, eps=0.01, optimizer='Adam', data_key='X', labels_key='labels', loss=<class 'torch.nn.modules.loss.CrossEntropyLoss'>)

Config for ClassifierTrainingPlan.

Attributes table

data_key
eps
labels_key
lr
optimizer
weight_decay

Methods table

to_kwargs()

Attributes

ClassifierTrainingPlanConfig.data_key: str = 'X'

ClassifierTrainingPlanConfig.eps: float = 0.01

ClassifierTrainingPlanConfig.labels_key: str = 'labels'

ClassifierTrainingPlanConfig.lr: float = 0.001

ClassifierTrainingPlanConfig.optimizer: Literal['Adam', 'AdamW'] = 'Adam'

ClassifierTrainingPlanConfig.weight_decay: float = 1e-06

Methods

ClassifierTrainingPlanConfig.to_kwargs()

Return type:

dict[str, Any]