sfujiwara

import tensorflow as tf
import optuna
import sklearn.datasets
from sklearn.model_selection import train_test_split


class TensorFlowPruningHook(tf.train.SessionRunHook):

    def __init__(self, trial, estimator, metric, is_higher_better, run_every_steps):
        self.trial = trial