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
Project details
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.
Source Distribution
Built Distribution
Hashes for scikeras-0.12.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 2d69a4a70118f8a0ffb1b2487b25f7c668f36ab83425fdf9a3271eb02da739ab |
|
MD5 | 7e6dd4b04c1452c603c27c631a1e7e76 |
|
BLAKE2b-256 | 5dfa9c1967952e7889d698f10ba8b6af79dfaa2e05178d97a79fbd9d1b44e589 |