2023-09-13 00:11:20 +02:00
|
|
|
|
|
|
|
|
from sklearn.linear_model import LogisticRegression
|
|
|
|
|
from quacc.baseline import kfcv
|
2023-09-14 01:52:19 +02:00
|
|
|
from quacc.dataset import get_spambase
|
2023-09-13 00:11:20 +02:00
|
|
|
|
|
|
|
|
|
|
|
|
|
class TestBaseline:
|
|
|
|
|
|
|
|
|
|
def test_kfcv(self):
|
2023-09-14 01:52:19 +02:00
|
|
|
train, _, _ = get_spambase()
|
2023-09-13 00:11:20 +02:00
|
|
|
c_model = LogisticRegression()
|
|
|
|
|
assert "f1_score" in kfcv(c_model, train)
|
