Prediction of the label can be made by calling the following method:
def predict(self, test_X): with self.tf_graph.as_default(): with tf.Session() as self.tf_session: self.tf_saver.restore(self.tf_session, self.model_path) feed = { self.input_data: test_X, self.keep_prob: 1 } return self.mod_y.eval(feed)