optuna.integration.TensorFlowPruningHook
- class optuna.integration.TensorFlowPruningHook(trial, estimator, metric, run_every_steps)[source]
TensorFlow SessionRunHook to prune unpromising trials.
See the example if you want to add a pruning hook to TensorFlow’s estimator.
- Parameters
Methods
after_run
(run_context, run_values)before_run
(run_context)begin
()