Neptune.ai tensorflow-keras integration library
Project description
Neptune + TensorFlow/Keras Integration
Experiment tracking, model registry, data versioning, and live model monitoring for TensorFlow/Keras trained models.
What will you get with this integration?
- Log, display, organize, and compare ML experiments in a single place
- Version, store, manage, and query trained models, and model building metadata
- Record and monitor model training, evaluation, or production runs live
- Collaborate with a team
What will be logged to Neptune?
- hyperparameters for every run,
- learning curves for losses and metrics during training,
- hardware consumption and stdout/stderr output during training,
- TensorFlow tensors as images to see model predictions live,
- training code and git commit information,
- model weights
- other metadata
Example charts in the Neptune UI with logged accuracy and loss
Resources
Example
# On the command line:
pip install tensorflow neptune-client neptune-tensorflow-keras
# In Python:
import neptune.new as neptune
from neptune.new.integrations.tensorflow_keras import NeptuneCallback
# Start a run
run = neptune.init(project="common/tf-keras-integration",
api_token="ANONYMOUS")
# Create a NeptuneCallback instance
neptune_cbk = NeptuneCallback(run=run, base_namespace="metrics")
# Pass the callback to model.fit()
model.fit(x_train, y_train,
epochs=5,
batch_size=64,
callbacks=[neptune_cbk])
# Stop the run
run.stop()
Support
If you got stuck or simply want to talk to us, here are your options:
- Check our FAQ page
- You can submit bug reports, feature requests, or contributions directly to the repository.
- Chat! When in the Neptune application click on the blue message icon in the bottom-right corner and send a message. A real person will talk to you ASAP (typically very ASAP),
- You can just shoot us an email at support@neptune.ai
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
Built Distribution
Close
Hashes for neptune_tensorflow_keras-1.2.0.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | afe5001bb34dc83b3ff8a4acb6213910e961b24536af2af8a61fb5085c6e0cb5 |
|
MD5 | 460ab4e0812a4a7cb14d5d6ef625a519 |
|
BLAKE2b-256 | 843a0a420f1e3151c7786f41254fe7cc774d48e5e09e727d954ac50307841cd8 |
Close
Hashes for neptune_tensorflow_keras-1.2.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | d603af52d16b2a769cdfa7820f9031ba6380ca7924bfa88bdf7882e1a6a549ea |
|
MD5 | 001b8074b367a84c300d6fbaa7b4e159 |
|
BLAKE2b-256 | ca63d362015bb37fd4ff6a7a7c38ac78d6dde58b2d743d597420fa6b4f3df58f |