def get_class(self, index): label = self.db.test.labels[index:index+1] return self.sess.run(tf.arg_max(label, 1))