A collection of functions to help you easily train and run Tensorflow Keras
Project description
Keras Toolkit
A collection of functions to help you easily train and run Tensorflow Keras
Quickstart
Install the library:
pip install keras-toolkit
You can now use it:
import keras_toolkit as kt
# kt reduces the number of lines from ~100 to ~3
strategy = kt.accelerator.auto_select(verbose=True)
decoder = kt.image.build_decoder(with_labels=True, target_size=(300, 300))
dtrain = kt.image.build_dataset(paths, labels, bsize=BATCH_SIZE, decode_fn=decoder)
with strategy.scope():
model = tf.keras.Sequential([...])
model.compile(...)
model.fit(...)
API References
Get the complete API reference here.
Acknowledgement
The auto_select_accelerator
was inspired by Martin Gorner's Kaggle notebook.
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
keras-toolkit-0.1.0rc4.tar.gz
(4.0 kB
view hashes)
Built Distribution
Close
Hashes for keras_toolkit-0.1.0rc4-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | b2388c2926df59bbe541643f12c684894e7346ad66c16aa5a39c75c874591eb7 |
|
MD5 | f8e4e084a04f218f90f377a51f832ef4 |
|
BLAKE2b-256 | 02dc1caedeebf4224aa3e56538348b8d862facf98f46d7c4a7b736959c579d65 |