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)],
)