skorch 一个基于pytorch封装兼容scikit-learn的神经网络库-python

【文件属性】：

文件名称：skorch 一个基于pytorch封装兼容scikit-learn的神经网络库-python

文件大小：2.82MB

文件格式：ZIP

更新时间：2024-07-08 12:01:59

机器学习

skorch 一个基于pytorch封装兼容scikit-learn的神经网络库 包装 PyTorch 的 scikit-learn 兼容神经网络库。 资源 文档 源代码示例 要查看更详细的示例，请查看此处。 import numpy as np from sklearn.datasets import make_classification from torch import nn import torch.nn.functional as F from skorch import NeuralNetClassifier X, y = make_classification(1000, 20, n_informative=10, random_state=0) X = X.astype( np.float32) y = y.astype(np.int64) class MyModule(nn.Module): def __init__(self, num_units=10, nonlin=F.relu): super(MyModule, self).__init__() self