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