from tensorflow.contrib.learn.python.learn.graph_actions import train

      self._check_inputs(features, targets)
      train_op, loss_op = self._get_train_ops(features, targets)
      return train(
          graph=g,
