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.0rc3.tar.gz
(4.1 kB
view hashes)
Built Distribution
Close
Hashes for keras_toolkit-0.1.0rc3-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 22a2b875a647f785a4df1c752108064f77cd5493279186bbac02def603af00a6 |
|
MD5 | 4922959559f8c743337fafe542880cf6 |
|
BLAKE2b-256 | ed7a9c1bbc19cc21bbbce25a24e4b04673ee3fc7838839e28ae5abeb40b1452b |