optuna.integration.TFKerasPruningCallback
- class optuna.integration.TFKerasPruningCallback(trial, monitor)[source]
tf.keras callback to prune unpromising trials.
This callback is intend to be compatible for TensorFlow v1 and v2, but only tested with TensorFlow v2.
See the example if you want to add a pruning callback which observes the validation accuracy.
- Parameters
Methods
on_epoch_end
(epoch[, logs])