def test_recall(self): rec = Recall() self.assertEqual(rec.type, "ranking") self.assertEqual(rec.name, "Recall@-1") self.assertEqual(1, rec.compute(np.asarray([1]), np.asarray([0]))) ground_truth = np.asarray([1, 0, 1]) # [1, 0, 1] rec_list = np.asarray([0, 2, 1]) # [1, 1, 1] self.assertEqual(1, rec.compute(ground_truth, rec_list)) ground_truth = np.asarray([0, 0, 1]) # [0, 0, 1] rec_list = np.asarray([1, 2, 0]) # [1, 1, 1] self.assertEqual(1, rec.compute(ground_truth, rec_list)) rec_2 = Recall(k=2) self.assertEqual(rec_2.k, 2) ground_truth = np.asarray([0, 0, 1]) # [0, 0, 1] rec_list = np.asarray([1, 2, 0]) # [1, 1, 1] self.assertEqual(1, rec_2.compute(ground_truth, rec_list))
def test_recall(): rec = Recall() assert rec.type == 'ranking' assert rec.name == 'Recall@-1' assert 1 == rec.compute(np.asarray([1]), np.asarray([0])) ground_truth = np.asarray([1, 0, 1]) # [1, 0, 1] rec_list = np.asarray([0, 2, 1]) # [1, 1, 1] assert 1 == rec.compute(ground_truth, rec_list) ground_truth = np.asarray([0, 0, 1]) # [0, 0, 1] rec_list = np.asarray([1, 2, 0]) # [1, 1, 1] assert 1 == rec.compute(ground_truth, rec_list) rec_2 = Recall(k=2) assert rec_2.k == 2 ground_truth = np.asarray([0, 0, 1]) # [0, 0, 1] rec_list = np.asarray([1, 2, 0]) # [1, 1, 1] assert 1 == rec_2.compute(ground_truth, rec_list)