训练包装机torch.nn.模块,提供类似scikitlearn的fit和predict接口。
torch-train的Python项目详细描述
火炬列车
训练包装材料torch.nn.模块,提供scikit learn-like fit和predict接口。在
安装
最简单的安装方法是通过pip
pip install torch-train
依赖性
火炬列车是Pythorch图书馆的延伸。 因此,当手动安装时,请确保安装了Pythorch。 通过pip:
^{pr2}$有关详细的安装指南,请参见https://pytorch.org/。在
文件
有关完整参考,请参见torch-train.readthedocs.io。在
- 项目
标签: