VertaHook¶
- class verta.integrations.tensorflow.VertaHook(*args: Any, **kwargs: Any)¶
TensorFlow Estimator hook that automates logging to Verta during model training.
This hook logs loss during training.
This hook has been verified to work with the TensorFlow 1.X API.
New in version 0.13.20.
- Parameters:
run (
ExperimentRun
) – Experiment Run tracking this model.every_n_steps (int, default 1000) – How often to log summary metrics.
Examples
from verta.integrations.tensorflow import VertaHook run = client.set_experiment_run() estimator.train( input_fn=train_input_fn, hooks=[VertaHook(run)], )