模板工具与锅炉板代码，用于在python中构建健壮的机器学习项目。
mltemplate的Python项目详细描述
ML模板
ML-template是一个易于使用的工具，可以自动执行大多数机器学习项目的锅炉板代码。在
该工具为机器学习项目创建了一个面向用户的项目体系结构。在
修改模板项目存储库中#TODO注释下的代码，以便轻松地使模板适应您的用例。在
如何使用它？在
- 将包安装为-
pip install mltemplate - 然后，只需从终端运行
mltempate，并按照提示操作
瞧！在
这将在当前文件夹中创建类似于-
template
├── Dockerfile.cpu
├── Dockerfile.gpu
├── Makefile
├── pyproject.toml
├── poetry.lock
├── notebooks
├── README.md
└── template
├── cli
│ ├── __init__.py
│ ├── __main__.py
│ ├── predict.py
│ └── train.py
├── __init__.py
├── models.py
├── datasets.py
└── transforms.py
你接下来要做的就是-
- 前往
template/datasets.py并修改创建一个新的数据集，该数据集将适用于您的用例 - 导航到
template/models.py并使用sota（或不是）体系结构创建一个新的模型类 - 在
template/transforms.py中添加诸如Normalizer、Denormalize等转换 - 按照
template/cli/train.py和template/cli/predict.py中的TODO步骤进行必要的更改
查看template目录中的README.md，了解如何使用张量板训练、预测和监控损失曲线。在
今后的工作
目前这个包只支持在^{
我们计划在将来支持tensorflow。在
许可证
版权所有©2020 Sowmya Yellapragada
根据麻省理工学院许可证（MIT）分发。在
- 项目
标签：
欢迎加入QQ群-->： 979659372
