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