scglue.models.scclue.SCCLUETrainer.fit

SCCLUETrainer.fit(data, val_split=None, batch_size=None, align_burnin=None, max_epochs=None, patience=None, reduce_lr_patience=None, random_seed=None, directory=None, plugins=None)

训练模型

参数:

• train_loader – Training data loader

• val_loader – Validation data loader

• max_epochs (typing.Optional[int]) – Maximal number of epochs

• random_seed (typing.Optional[int]) – 随机种子

• directory (typing.Optional[os.PathLike]) – Training directory

• plugins (typing.Optional[typing.List[scglue.models.base.TrainingPlugin]]) – Optional list of training plugins

返回类型:

None