Sync App files
Browse files- test/test_model.py +1 -1
test/test_model.py
CHANGED
|
@@ -14,7 +14,7 @@ model = joblib.load("./model.pkl")
|
|
| 14 |
def test_auc():
|
| 15 |
|
| 16 |
preds = model.predict(X_test)
|
| 17 |
-
f1 = f1_score(y_test, preds)
|
| 18 |
assert f1 > 0.85, "f1_score is below acceptable threshold"
|
| 19 |
|
| 20 |
def test_missing_values():
|
|
|
|
| 14 |
def test_auc():
|
| 15 |
|
| 16 |
preds = model.predict(X_test)
|
| 17 |
+
f1 = f1_score(y_test, preds, average="macro")
|
| 18 |
assert f1 > 0.85, "f1_score is below acceptable threshold"
|
| 19 |
|
| 20 |
def test_missing_values():
|