文件名称:SKORCH-PyTorch-Wrapper:此存储库显示了SKORCH使用scikit-learn框架的不同功能来训练PyTorch模型的可用性的示例
文件大小:238KB
文件格式:ZIP
更新时间:2024-02-24 19:17:56
wrapper neural-network scikit-learn pytorch wrapper-library
SKORCH:使用Scikit-Learn包装器训练的PyTorch模型 该存储库显示了SKORCH使用scikit-learn框架的不同功能来训练PyTorch模型的可用性的示例。 如果您想了解有关如何创建此模型的详细信息,请查看以下非常清晰而详细的说明: 目录 1.模型 该存储库的想法是展示如何使用一些SKorch功能来训练PyTorch模型。 在这种情况下,创建了一个神经网络来对葡萄酒数据集进行分类。 为了更好地理解什么是SKorch,请看以下图片: 2.档案 数据:在这里您将找到葡萄酒数据集 src :它包含两个文件data_loader.py和model.py 。 文件data_l
【文件预览】:
SKORCH-PyTorch-Wrapper-master
----data()
--------wines.csv(11KB)
----src()
--------__init__.py(63B)
--------data_loader.py(622B)
--------model.py(615B)
----Pipfile.lock(14KB)
----README.md(4KB)
----Pipfile(195B)
----.vscode()
--------settings.json(103B)
----main.py(3KB)
----img()
--------skorch2.jpg(240KB)