()
| 91 | |
| 92 | |
| 93 | def test_gbm(): |
| 94 | model = GradientBoostingClassifier( |
| 95 | n_estimators=25, max_depth=3, max_features=5, learning_rate=0.1 |
| 96 | ) |
| 97 | model.fit(X_train, y_train) |
| 98 | predictions = model.predict(X_test) |
| 99 | assert roc_auc_score(y_test, predictions) >= 0.95 |
| 100 | |
| 101 | |
| 102 | def test_naive_bayes(): |
nothing calls this directly
no test coverage detected