Skip to main content

用于 Keras 的 Scikit-Learn API 包装器。

项目描述

用于 Keras 的 Scikit-Learn 包装器

构建状态 覆盖状态 文档

用于 Keras 模型的 Scikit-Learn 兼容包装器。

为什么选择 SciKeras

SciKeras 源自tf.keras.wrappers.scikit_learn. 原始的 TensorFlow (TF) 包装器没有得到积极维护,在未来的版本中删除。

可以在我们的 迁移指南中找到与 TF 包装器相比的优势和差异的概述。

安装

这个包在 PyPi 上可用:

# Normal tensorflow
pip install scikeras[tensorflow]

# or tensorflow-cpu
pip install scikeras[tensorflow-cpu]

SciKeras 将 TensorFlow 打包为可选依赖项,因为有多种 TensorFlow 可用(tensorflow、、tensorflow-cpu等)。取决于其中一个特别不允许使用另一个,这就是为什么它们需要是可选的。

pip install scikeras[tensorflow]基本上等同于pip install scikeras tensorflow 并且只是为了方便而提供。您也可以使用 仅安装 SciKeras pip install scikeras,但您需要在运行时安装 tensorflow 版本,否则 SciKeras 会在您尝试导入时抛出错误。

当前版本的 SciKeras 依赖于scikit-learn>=1.0.0TensorFlow>=2.7.0.

迁移自tf.keras.wrappers.scikit_learn

请参阅我们文档的迁移部分。

文档

文档可在https://www.adriangb.com/scikeras/获得。

贡献

请参阅CONTRIBUTING.md

项目详情


下载文件

下载适用于您平台的文件。如果您不确定要选择哪个,请了解有关安装包的更多信息。

源分布

scikeras-0.9.0.tar.gz (26.1 kB 查看哈希

已上传 source

内置分布

scikeras-0.9.0-py3-none-any.whl (27.8 kB 查看哈希

已上传 py3