def test(self, x, a, t, b, learning_rate=1e-2, momentum=0.9): feed_dict = utils.construct_feed_dict(x, a, t, b, learning_rate, momentum, self.placeholders) acc, pred, probabilities, tags = self.sess.run([ self.accuracy, self.pred, self.probabilities, self.correct_prediction ], feed_dict=feed_dict) return acc, pred, probabilities, tags
def train(self, x, a, t, b, learning_rate=1e-2, momentum=0.9): feed_dict = utils.construct_feed_dict(x, a, t, b, learning_rate, momentum, self.placeholders) outs = self.sess.run( [self.train_op, self.loss, self.accuracy, self.pred, self.probabilities], feed_dict=feed_dict) loss = outs[1] acc = outs[2] pred = outs[3] prob = outs[4] return loss, acc, pred, prob