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
Get the complete API reference here.
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(...)
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.0rc5.tar.gz
(4.5 kB
view hashes)
Built Distribution
Close
Hashes for keras_toolkit-0.1.0rc5-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | bdf09ff6e3f1b0d199fb386616ffebd2128b158b822352557958ae5511781405 |
|
MD5 | b66c8651fcd470243619bb9e717f8e8d |
|
BLAKE2b-256 | 7f83acd6a2905354ef34338de9e28044d295a90ffb7a44ca52e6e9fa3191f508 |