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:

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

Please see the migration section of our documentation.

Documentation is available at https://www.adriangb.com/scikeras/.

See CONTRIBUTING.md