optuna.integration.TensorBoardCallback

class optuna.integration.TensorBoardCallback(dirname, metric_name)[source]

Callback to track Optuna trials with TensorBoard.

This callback adds relevant information that is tracked by Optuna to TensorBoard.

See the example.

Parameters
  • dirname (str) – Directory to store TensorBoard logs.

  • metric_name (str) – Name of the metric. Since the metric itself is just a number, metric_name can be used to give it a name. So you know later if it was roc-auc or accuracy.

Note

Added in v2.0.0 as an experimental feature. The interface may change in newer versions without prior notice. See https://github.com/optuna/optuna/releases/tag/v2.0.0.