scglue.models.base.Trainer.fit

Trainer.fit(train_loader, val_loader=None, max_epochs=100, random_seed=0, directory=None, plugins=None)

训练模型

参数:

• train_loader (typing.Iterable) – Training data loader

• val_loader (typing.Optional[typing.Iterable]) – Validation data loader

• max_epochs (int) – Maximal number of epochs

• random_seed (int) – 随机种子

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

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

返回类型:

None