classifier.py

class Model:
  def __init__(self, input, labels):
    """Classifier model
    Args:
      input: Input tensor of size (None, input_dims)
      label: Label tensor of size (None, 1). 
        Should be of type tf.int32.
    """
    prediction = self.make_network(input)
    # Prediction size is (None, 1).
    self.loss = tf.nn.softmax_cross_entropy_with_logits(
        logits=prediction, labels=labels)
    self.train_op = tf.train.AdamOptimizer().minimize(self.loss)