from tensorflow.contrib import layers

      raise ValueError("n_classes should be greater than 1. Given: {}".format(
          n_classes))
    target_column = layers.multi_class_target(
        n_classes=n_classes,
