'Predict the labels for the test set.
Parameters
test_X : array_like, shape (n_samples, n_features)
Test data.
Returns
array_like, shape (n_samples,) : predicted labels.'

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)
