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 – Directory to store TensorBoard logs.
metric_name – 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.