def test_expected_data_uncertainty_with_logits(self):
_, _, output = um.model_uncertainty(logits=self.logits)
msg = "Data uncertainty not almost equal what is expected."
self.assertTensorsAlmostEqual(self.expected_data_uncertainty,
output,
msg=msg)
def test_wrong_logits_shape(self):
logits_wrong = tf.reduce_mean(self.logits, -1)
with self.assertRaises(ValueError):
um.model_uncertainty(logits=logits_wrong)
def test_total_uncertainty_with_logits(self):
_, output, _ = um.model_uncertainty(logits=self.logits)
msg = "Total uncertainty not almost equal what is expected."
self.assertTensorsAlmostEqual(self.total_uncertainty, output, msg=msg)
