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:
# Normal tensorflow
pip install scikeras[tensorflow]
# or tensorflow-cpu
pip install scikeras[tensorflow-cpu]
SciKeras packages TensorFlow as an optional dependency because there are
several flavors of TensorFlow available (tensorflow, tensorflow-cpu, etc.).
Depending on one of them in particular disallows the usage of the other, which is why
they need to be optional.
pip install scikeras[tensorflow] is basically equivalent to pip install scikeras tensorflow
and is offered just for convenience. You can also install just SciKeras with
pip install scikeras, but you will need a version of tensorflow installed at
runtime or SciKeras will throw an error when you try to import it.
The current version of SciKeras depends on scikit-learn>=1.0.0 and TensorFlow>=2.7.0.
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.
