removed baseline tests

This commit is contained in:
Lorenzo Volpi 2023-12-08 00:44:16 +01:00
parent de48da638a
commit fbe5d05c3e
2 changed files with 0 additions and 12 deletions

View File

@ -1,12 +0,0 @@
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)