def test_desp_proba(): pool_classifiers, X_dsel, y_dsel, X_test, y_test = setup_classifiers() desp = DESP(pool_classifiers, DFP=True) desp.fit(X_dsel, y_dsel) probas = desp.predict_proba(X_test) expected = np.load('deslib/tests/expected_values/desp_proba_DFP.npy') assert np.allclose(probas, expected)
def test_desp_proba(knn_methods): pool_classifiers, X_dsel, y_dsel, X_test, y_test = setup_classifiers() desp = DESP(pool_classifiers, knn_classifier=knn_methods) desp.fit(X_dsel, y_dsel) probas = desp.predict_proba(X_test) expected = np.load( 'deslib/tests/expected_values/desp_proba_integration.npy') assert np.allclose(probas, expected)