Modify weights test to make it significative
@@ -275,7 +275,7 @@ class TverskyTest(tf.test.TestCase):
predicted = tf.constant(
[[0, 10], [10, 0], [10, 0], [10, 0]],
dtype=tf.float32, name='predicted')
- labels = tf.constant([[1, 0, 0, 0]], dtype=tf.int64, name='labels')
+ labels = tf.constant([[1, 0, 1, 0]], dtype=tf.int64, name='labels')
predicted, labels = [tf.expand_dims(x, axis=0) for x in (predicted, labels)]
test_loss_func = LossFunction(2,
