Neptune.ai tensorflow-keras integration library
Project description
Neptune: Tensorflow and Keras integration
See the official docs.
Minimal example:
import tensorflow as tf
import neptune.new as neptune
from neptune.new.integrations.tensorflow_keras import NeptuneCallback
run = neptune.init(project='common/tf-keras-integration', api_token='ANONYMOUS')
mnist = tf.keras.datasets.mnist
(x_train, y_train),(x_test, y_test) = mnist.load_data()
x_train, x_test = x_train / 255.0, x_test / 255.0
model = tf.keras.models.Sequential([
tf.keras.layers.Flatten(),
tf.keras.layers.Dense(256, activation=tf.keras.activations.relu),
tf.keras.layers.Dropout(0.5),
tf.keras.layers.Dense(10, activation=tf.keras.activations.softmax)
])
optimizer = tf.keras.optimizers.SGD(lr=0.005, momentum=0.4,)
model.compile(optimizer=optimizer,
loss='sparse_categorical_crossentropy',
metrics=['accuracy'])
neptune_cbk = NeptuneCallback(run=run, base_namespace='metrics')
model.fit(x_train, y_train,
epochs=5,
batch_size=64,
callbacks=[neptune_cbk])
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
Close
Hashes for neptune-tensorflow-keras-0.9.5.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | e2908991b322a3ee7cfe02159497c472883c0b114a05d2881623594538deb949 |
|
MD5 | d52562b8cc18392d3692e5209d591144 |
|
BLAKE2b-256 | 140fb1f9dbba46935136ff25486aac3cd2c36025b84931e9d31b45a653f81222 |