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