Пример #1
0
def test_standard():
    y_targ = [0, 0, 0, 1, 1, 1]
    y_pred = [0, 1, 1, 0, 1, 1]

    y = accuracy_score(y_targ, y_pred, method='standard')
    y_1 = accuracy_score(y_targ, y_pred, method='binary', normalize=False)
    assert_almost_equal(y, float(3) / 6, decimal=4)
    assert y_1 == 3
Пример #2
0
def test_multiclass_binary():
    y_targ = [0, 0, 0, 1, 1, 1, 2, 2, 2]
    y_pred = [1, 0, 0, 0, 1, 2, 0, 2, 2]

    y_1 = accuracy_score(y_targ, y_pred, method='binary', pos_label=1)
    y_2 = accuracy_score(y_targ, y_pred, method='binary', pos_label=2)
    y_3 = accuracy_score(y_targ,
                         y_pred,
                         method='binary',
                         pos_label=1,
                         normalize=False)

    assert_almost_equal(y_2, float(7) / 9, decimal=4)
    assert_almost_equal(y_1, float(6) / 9, decimal=4)
    assert y_3 == 6
Пример #3
0
def test_average():
    y_targ = np.array([0, 0, 0, 1, 1, 1, 1, 1, 2, 2])
    y_pred = np.array([0, 1, 1, 0, 1, 1, 2, 2, 2, 2])

    y = accuracy_score(y_targ, y_pred, method='average')
    assert_almost_equal(y, float(2) / 3, decimal=4)
Пример #4
0
def test_balanced_binary():
    y_targ = np.array([0, 0, 0, 1, 1, 1, 1, 1, 1, 0])
    y_pred = np.array([0, 1, 1, 0, 1, 1, 1, 1, 1, 1])

    y = accuracy_score(y_targ, y_pred, method='balanced')
    assert_almost_equal(y, 0.542, decimal=3)
Пример #5
0
def test_balanced_multiclass():
    y_targ = np.array([0, 0, 0, 1, 1, 1, 1, 1, 2, 2])
    y_pred = np.array([0, 1, 1, 0, 1, 1, 2, 2, 2, 2])

    y = accuracy_score(y_targ, y_pred, method='balanced')
    assert_almost_equal(y, 0.578, decimal=3)