Exemplo n.º 1
0
    def test_mae(self):
        mae = MAE()

        self.assertEqual(mae.type, 'rating')
        self.assertEqual(mae.name, 'MAE')

        self.assertEqual(0, mae.compute(np.asarray([0]), np.asarray([0])))
        self.assertEqual(1, mae.compute(np.asarray([0, 1]), np.asarray([1, 0])))
        self.assertEqual(2, mae.compute(np.asarray([0, 1]), np.asarray([2, 3]), np.asarray([1, 3])))
Exemplo n.º 2
0
def test_mae():
    mae = MAE()

    assert mae.type == 'rating'
    assert mae.name == 'MAE'

    assert 0 == mae.compute(np.asarray([0]), np.asarray([0]))
    assert 1 == mae.compute(np.asarray([0, 1]), np.asarray([1, 0]))
    assert 2 == mae.compute(np.asarray([0, 1]), np.asarray([2, 3]),
                            np.asarray([1, 3]))