|
|
|
from sklearn.linear_model import LogisticRegression
|
|
from quacc.baseline import kfcv
|
|
from quacc.dataset import get_spambase
|
|
|
|
|
|
class TestBaseline:
|
|
|
|
def test_kfcv(self):
|
|
train, _, _ = get_spambase()
|
|
c_model = LogisticRegression()
|
|
assert "f1_score" in kfcv(c_model, train) |