13 lines
402 B
Python
13 lines
402 B
Python
from sklearn.linear_model import LogisticRegression
|
|
|
|
from quacc.dataset import Dataset
|
|
from quacc.evaluation.baseline import kfcv
|
|
|
|
|
|
class TestBaseline:
|
|
def test_kfcv(self):
|
|
spambase = Dataset("spambase", n_prevalences=1).get_raw()
|
|
c_model = LogisticRegression()
|
|
c_model.fit(spambase.train.X, spambase.train.y)
|
|
assert "f1_score" in kfcv(c_model, spambase.validation)
|