Scikit-Learn Wrapper for Keras
Scikit-Learn compatible wrappers for Keras Models.
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.
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 (
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
Please see the migration section of our documentation.
Documentation is available at https://www.adriangb.com/scikeras/.