def calc_errors(self, output):
Z = output['numbers']
N = tf.reduce_sum(tf.cast(tf.greater(Z, 0), tf.float32), 1)
tgt = output[self.target]
pred = output[self.prediction]
if self.idx is not None:
tgt = tgt[:, self.idx]
pred = pred[:, self.idx]
return ((tgt - pred) / N) ** 2
评论列表
文章目录