Ejemplo n.º 1
0
def test_ordinal_mcrank_set_estimator_params():
    gb = GradientBoostingClassifier(n_estimators=5,
                                    loss="deviance",
                                    random_state=0)
    mc = OrdinalMcRank(gb)
    mc.set_params(estimator__n_estimators=10)
    assert_equal(gb.n_estimators, 10)
Ejemplo n.º 2
0
def test_ordinal_mcrank():
    gb = GradientBoostingClassifier(n_estimators=10,
                                    loss="deviance",
                                    random_state=0)
    mc = OrdinalMcRank(gb)
    mc.fit(X, y)
    assert_almost_equal(mc.score(X, y), 48.62, 2)
Ejemplo n.º 3
0
def test_ordinal_mcrank_set_estimator_params():
    gb = GradientBoostingClassifier(n_estimators=5,
                                    loss="deviance",
                                    random_state=0)
    mc = OrdinalMcRank(gb)
    mc.set_params(estimator__n_estimators=10)
    assert_equal(gb.n_estimators, 10)
Ejemplo n.º 4
0
def test_ordinal_mcrank():
    gb = GradientBoostingClassifier(n_estimators=10,
                                    loss="deviance",
                                    random_state=0)
    mc = OrdinalMcRank(gb)
    mc.fit(X, y)
    assert_almost_equal(mc.score(X, y), 48.62, 2)
Ejemplo n.º 5
0
def test_ordinal_mcrank_warm_start():
    gb = GradientBoostingClassifier(n_estimators=5,
                                    loss="deviance",
                                    warm_start=True,
                                    random_state=0)

    mc = OrdinalMcRank(gb)
    mc.fit(X, y)
    assert_almost_equal(mc.score(X, y), 56.35, 2)

    mc.set_params(estimator__n_estimators=10)
    mc.fit(X, y)
    assert_almost_equal(mc.score(X, y), 48.62, 2)
Ejemplo n.º 6
0
def test_ordinal_mcrank_warm_start():
    gb = GradientBoostingClassifier(n_estimators=5,
                                    loss="deviance",
                                    warm_start=True,
                                    random_state=0)

    mc = OrdinalMcRank(gb)
    mc.fit(X, y)
    assert_almost_equal(mc.score(X, y), 56.35, 2)

    mc.set_params(estimator__n_estimators=10)
    mc.fit(X, y)
    assert_almost_equal(mc.score(X, y), 48.62, 2)