Keras的Scikit-Learn封装
Keras模型的Scikit-Learn兼容封装。
为什么选择SciKeras
SciKeras源自并兼容现已弃用/移除的tf.keras.wrappers.scikit_learn
接口。
与TF封装相比的差异概述可以在我们的迁移指南中找到。
安装
该包可在PyPi上获取:
# Tensorflow
pip install scikeras[tensorflow]
请注意,pip install scikeras[tensorflow]
基本等同于pip install scikeras tensorflow
,提供这种方式只是为了方便。你也可以只安装SciKeras,使用pip install scikeras
,但在运行时你需要安装tensorflow的某个版本,否则当你尝试导入SciKeras时会抛出错误。
当前版本的SciKeras依赖于scikit-learn>=1.4.1post1
和Keras>=3.2.0
。
从keras.wrappers.scikit_learn
迁移
请参阅我们文档中的迁移部分。
文档
文档可在https://www.adriangb.com/scikeras/获取。