def __init__(self, keras_model, graph, sess):
    self.keras_model = keras_model
    self._graph = graph
    self._sess = sess
def __init__(self, keras_model, graph, sess):
    self.keras_model = keras_model
    self._graph = graph
    self._sess = sess