def loss(self, labels):
    return tf.losses.mean_squared_error(labels=labels.action, predictions=self._predictions)
def loss(self, labels):
    return tf.losses.mean_squared_error(labels=labels.action, predictions=self._predictions)