tf.compat.v1.data.get_output_classes