skorch
模型生态skorch 是一个兼容 scikit-learn 的神经网络库,把 PyTorch 模型包装成标准 estimator,可直接用 fit、predict 等接口训练。典型场景是用 scikit-learn 的交叉验证和超参数搜索调 PyTorch 模型,无需手写训练循环。
查看 GitHub 原始描述(英文)
A scikit-learn compatible neural network library that wraps PyTorch
按钮会在新标签页打开,本站不代理、不镜像仓库内容。