20 lines
622 B
Python
20 lines
622 B
Python
|
|
from sklearn.linear_model import LogisticRegression
|
|
from quacc.baseline import kfcv, trust_score
|
|
from quacc.dataset import get_spambase
|
|
|
|
|
|
class TestBaseline:
|
|
|
|
def test_kfcv(self):
|
|
train, validation, _ = get_spambase()
|
|
c_model = LogisticRegression()
|
|
c_model.fit(train.X, train.y)
|
|
assert "f1_score" in kfcv(c_model, validation)
|
|
|
|
def test_trust_score(self):
|
|
train, validation, test = get_spambase()
|
|
c_model = LogisticRegression()
|
|
c_model.fit(train.X, train.y)
|
|
trustscore = trust_score(c_model, train, test)
|
|
assert len(trustscore) == len(test.y) |