def test_fista_multiclass_l1_no_line_search(): for data in (mult_dense, mult_csr): clf = FistaClassifier(max_iter=500, penalty="l1", multiclass=True, max_steps=0) clf.fit(data, mult_target) assert_almost_equal(clf.score(data, mult_target), 0.95, 2)
def test_fista_multiclass_l1l2_log_margin(): for data in (mult_dense, mult_csr): clf = FistaClassifier(max_iter=500, penalty="l1/l2", loss="log_margin", multiclass=True) clf.fit(data, mult_target) assert_almost_equal(clf.score(data, mult_target), 0.95)
def test_fista_multiclass_trace(): for data in (mult_dense, mult_csr): clf = FistaClassifier(max_iter=100, penalty="trace", multiclass=True) clf.fit(data, mult_target) assert_almost_equal(clf.score(data, mult_target), 0.98, 2)
def test_fista_bin_l1_no_line_search(): for data in (bin_dense, bin_csr): clf = FistaClassifier(max_iter=500, penalty="l1", max_steps=0) clf.fit(data, bin_target) assert_almost_equal(clf.score(data, bin_target), 1.0, 2)