removed baseline tests
This commit is contained in:
parent
de48da638a
commit
fbe5d05c3e
Binary file not shown.
|
@ -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)
|
|
Loading…
Reference in New Issue