def testNumRelevantSparse(self):
with self.test_session():
labels = tf.SparseTensorValue(
indices=(
(0, 0, 0), (0, 0, 1),
(0, 1, 0), (0, 1, 1), (0, 1, 2),
# (0, 2) missing
(1, 0, 0), (1, 0, 1), (1, 0, 2),
(1, 1, 0),
(1, 2, 0),
# (2, 0) missing
(2, 1, 0), (2, 1, 1),
(2, 2, 0)),
values=(1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13),
shape=(3, 3, 3))
self.assertAllEqual(
((1, 1, 0), (1, 1, 1), (0, 1, 1)),
metric_ops.num_relevant(labels, k=1).eval())
self.assertAllEqual(
((2, 2, 0), (2, 1, 1), (0, 2, 1)),
metric_ops.num_relevant(labels, k=2).eval())
label_lengths = ((2, 3, 0), (3, 1, 1), (0, 2, 1))
self.assertAllEqual(
label_lengths, metric_ops.num_relevant(labels, k=3).eval())
self.assertAllEqual(
label_lengths, metric_ops.num_relevant(labels, k=999).eval())
评论列表
文章目录