コード例 #1
0
ファイル: test_dual_cd.py プロジェクト: nagyistge/lightning
def test_fit_linear_binary():
    for loss in ("l1", "l2"):
        clf = DualLinearSVC(loss=loss, random_state=0, max_iter=100)
        clf.fit(bin_dense, bin_target)
        y_pred = clf.decision_function(bin_dense)

        clf2 = DualSVC(loss=loss, random_state=0, max_iter=100)
        clf2.fit(bin_dense, bin_target)
        y_pred2 = clf2.decision_function(bin_dense)

        assert_array_almost_equal(y_pred, y_pred2)
コード例 #2
0
ファイル: test_dual_cd.py プロジェクト: nagyistge/lightning
def test_warm_start():
    clf = DualSVC(warm_start=True, loss="l1", kernel="linear", random_state=0,
                  max_iter=100)
    clf2 = DualLinearSVC(warm_start=True, loss="l1", random_state=0,
                         max_iter=100)

    for C in (0.1, 0.2):
        clf.C = C
        clf2.C = C

        clf.fit(bin_dense, bin_target)
        y_pred = clf.decision_function(bin_dense)
        clf2.fit(bin_dense, bin_target)
        y_pred2 = clf2.decision_function(bin_dense)

        assert_array_almost_equal(y_pred, y_pred2)