SKORCH PyTorch Wrapper:此存储库显示了SKORCH使用scikit learn框架的不同功能来训练PyTorch模型的可用性的示例 源码
SKORCH:使用Scikit-Learn包装器训练的PyTorch模型 该存储库显示了SKORCH使用scikit-learn框架的不同功能来训练PyTorch模型的可用性的示例。 如果您想了解有关如何创建此模型的详细信息,请查看以下非常清晰而详细的说明: 目录 1.模型 该存储库的想法是展示如何使用一些SKorch功能来训练PyTorch模型。 在这种情况下,创建了一个神经网络来对葡萄酒数据集进行分类。 为了更好地理解什么是SKorch,请看以下图片: 2.档案 数据:在这里您将找到葡萄酒数据集 src :它包含两个文件data_loader.py和model.py 。 文件data_l
下载地址
用户评论