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.9.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | c1586ff4da97d0bc5836ddbdd722c216191f35e4072205183a8a273b96e0b157 |
|
MD5 | ce16b137d89f74c8f1f3fd1aacc5eb12 |
|
BLAKE2b-256 | aeaffe32a0ff53298ee68105065baf1746e70324f79bfab8d8698afb228d5bd7 |