Esempio n. 1
0
def test_corner_preds_horiz_micro(y_true_horiz, y_pred_horiz):
    assert pytest.approx(4 / 9) == f1_loss(y_true_horiz, y_pred_horiz, 'micro')
Esempio n. 2
0
def test_corner_preds_vert_micro(y_true_vert, y_pred_vert):
    assert pytest.approx(4 / 9) == f1_loss(y_true_vert, y_pred_vert, 'micro')
Esempio n. 3
0
def test_corner_preds_horiz_macro(y_true_horiz, y_pred_horiz):
    assert pytest.approx(11 / 24) == f1_loss(y_true_horiz, y_pred_horiz,
                                             'macro')
Esempio n. 4
0
def test_true_one_pred_half(one, half):
    assert pytest.approx(1 / 3) == f1_loss(one, half, 'macro')
Esempio n. 5
0
def test_corner_preds_vert_macro(y_true_vert, y_pred_vert):
    assert pytest.approx(9 / 20) == f1_loss(y_true_vert, y_pred_vert, 'macro')
Esempio n. 6
0
def test_true_zero_pred_half(zero, half):
    assert pytest.approx(1.0) == f1_loss(zero, half, 'micro')
Esempio n. 7
0
def test_true_one_pred_one(one):
    assert pytest.approx(0.0) == f1_loss(one, one, 'micro')
Esempio n. 8
0
def test_true_one_pred_zero(one, zero):
    assert pytest.approx(1.0) == f1_loss(one, zero, 'macro')
Esempio n. 9
0
def test_true_zero_pred_one(zero, one):
    assert pytest.approx(1.0) == f1_loss(zero, one, 'micro')
Esempio n. 10
0
def test_true_zero_pred_zero(zero):
    assert pytest.approx(1.0) == f1_loss(zero, zero, 'macro')