Scikit-Learn API wrapper for Keras.
Project description
Scikit-Learn Wrapper for Keras
Scikit-Learn compatible wrappers for Keras Models.
Why SciKeras
SciKeras is derived from and API compatible with tf.keras.wrappers.scikit_learn. The original TensorFlow (TF) wrappers are not actively maintained,
and will be removed in a future release.
An overview of the advantages and differences as compared to the TF wrappers can be found in our migration guide.
Installation
This package is available on PyPi:
pip install scikeras
The only dependencies are scikit-learn>=0.22 and TensorFlow>=2.4.0.
You will need to manually install TensorFlow; due to TensorFlow's packaging it is not a direct dependency of SciKeras. You can do this by running:
pip install tensorflow
Migrating from tf.keras.wrappers.scikit_learn
Please see the migration section of our documentation.
Documentation
Documentation is available at https://www.adriangb.com/scikeras/.
Contributing
See CONTRIBUTING.md
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
