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.data.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.0rc2.tar.gz
(4.1 kB
view hashes)
Built Distribution
Close
Hashes for keras_toolkit-0.1.0rc2-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | d5434225030649976812e495f6aedff3568a6a02dee40c9f14346e59cee2af93 |
|
MD5 | ea893008d502d4efe5e87106c6a165b0 |
|
BLAKE2b-256 | 53842654dc98b537c10cd6e373d30eb5d4d1b2e48a0d627d97197a45a15dc5d9 |