def __init__(self, model=None, graph=None, current_epoch=0, featurizer=None, max_history=None):
    import keras
    super(KerasPolicy, self).__init__(featurizer, max_history)
    if (KerasPolicy.is_using_tensorflow() and (not graph)):
        self.graph = keras.backend.tf.get_default_graph()
    else:
        self.graph = graph
    self.model = model
    self.current_epoch = current_epoch
def __init__(self, model=None, graph=None, current_epoch=0, featurizer=None, max_history=None):
    import keras
    super(KerasPolicy, self).__init__(featurizer, max_history)
    if (KerasPolicy.is_using_tensorflow() and (not graph)):
        self.graph = keras.backend.tf.get_default_graph()
    else:
        self.graph = graph
    self.model = model
    self.current_epoch = current_epoch