def testSingleUpdateWithErrorAndWeights1(self):
np_predictions = np.matrix(('1 0 0;'
'0 0 -1;'
'1 0 0'))
np_labels = np.matrix(('1 0 0;'
'0 0 1;'
'0 1 0'))
predictions = tf.constant(np_predictions, shape=(3, 1, 3), dtype=tf.float32)
labels = tf.constant(np_labels, shape=(3, 1, 3), dtype=tf.float32)
weights = tf.constant([1, 0, 0], shape=(3, 1, 1), dtype=tf.float32)
error, update_op = metrics.streaming_mean_cosine_distance(
predictions, labels, dim=2, weights=weights)
with self.test_session() as sess:
sess.run(tf.initialize_local_variables())
self.assertEqual(0, sess.run(update_op))
self.assertEqual(0, error.eval())
评论列表
文章目录