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)
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)
