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 loaderval_loader (
typing.Optional
[typing.Iterable
]) – Validation data loadermax_epochs (
int
) – Maximal number of epochsrandom_seed (
int
) – 随机种子directory (
typing.Optional
[os.PathLike
]) – Training directoryplugins (
typing.Optional
[typing.List
[scglue.models.base.TrainingPlugin
]]) – Optional list of training plugins
- 返回类型: