pytorch默认初始化_使用Scikit-Learn训练PyTorch模型

你是否想过会有一种工具，它用一种简单易行的方式来训练和评估PyTorch模型？这个工具是存在的，它就是SKORCH，一个用于训练PyTorch模型的Scikit-Learn包装器。

本文将讨论SKORCH是什么、它的组件以及包装PyTorch模型来训练和评估它有多容易。本文将分为以下几个部分：

•什么是SKORCH

•PyTorch模型

•与SKORCH一起训练

我们开始吧！

什么是SKORCH？

SKORCH是Scikit-Learn和PyTorch的结合，换句话说，SKORCH是一个用于训练、调优和优化PyTorch模型的包装器。SKORCH是2017年推出的一个开源库，SKORCH的出现是为了将PyTorch和Scikit-Learn框架的巨大优点结合起来并加以提升。

PyTorch是开发神经网络模型最常用的框架之一，但有时它会变得有些不切实际。

SKORCH试图简化PyTorch模型训练阶段的各种过程。PyTorch模型的训练模块通常是在一个或多个功能中开发的，但是，当需要对模型进行评估或优化以找到最佳参数时，需要开发额外的功能。SKORCH简化了这些过程，因为它基于scikit-learn的包装器，扩展了这些过程。

SKORCH宣称的哲学是：“做一个Scikit-Learn API，不隐藏PyTorch，不重新发明轮子”。

在下图中，我们可以看到SKORCH结合了PyTorch和scikit-learn的功能，正如我们所看到的，它从PyTorch方面继承了原型模型和处理数据集的功能；另一方面，scikit-learn的功能被扩展到能够训练、评估、调整和优化机器学习模型。这种组合使SKORCH成为一个强大的工具。

再者，我们可以将SKORCH视为keras api的“等价物”，它是从Tensorflow扩展到加速、简化和加速神经网络模型的原型。在这种情况下，SKORCH将作为PyTorch模型的训练、调优和优化阶段的原型工具。

太好了，到目前为止，我们已经知道了SKORCH是什么，它的组件是什么，以及使用它的优势，接下来通过一个例子更好地理解它是如何工作的，让我们开始吧！

PyTorch模型

为了了解SKORCH在训练PyTorch模型时是如何工作的，我们将创建一个神经网络来预测著名葡萄酒数据集。因此，首先我们将根据上述数据集创建一个简单的葡萄酒分类模型&#x