def test_functional_ignore_index_with_valid_index(self): actual = topk(self.predictions, self.targets, 3, ignore_index=1) self.assertEqual((), actual.shape) np.testing.assert_almost_equal(40., actual)
def test_functional_ignore_index_with_different_value(self): actual = topk(self.predictions, self.targets, 3, ignore_index=-1) self.assertEqual((), actual.shape) self.assertAlmostEqual(50., float(actual), places=5)
def test_functional_standard(self): actual = topk(self.predictions, self.targets, 3) self.assertEqual((), actual.shape) np.testing.assert_almost_equal(self.top_k_acc, actual)
def test_functional_on_gpu(self): actual = topk(self.predictions.cuda(), self.targets.cuda(), 3) self.assertEqual((), actual.shape) np.testing.assert_almost_equal(self.top_k_acc, actual.cpu(), decimal=5)
def test_functional_none(self): actual = topk(self.predictions, self.targets, 3, reduction='none') self.assertEqual(self.targets.shape, actual.shape) self.assertTrue(torch.all(self.top_k_acc_none == actual))
def test_functional_sum(self): actual = topk(self.predictions, self.targets, 3, reduction='sum') self.assertEqual((), actual.shape) np.testing.assert_almost_equal(self.top_k_acc * len(self.predictions), actual, decimal=0)