def test_kne_proba(): pool_classifiers, X_dsel, y_dsel, X_test, y_test = setup_classifiers() kne = KNORAE(pool_classifiers, DFP=True) kne.fit(X_dsel, y_dsel) probas = kne.predict_proba(X_test) expected = np.load('deslib/tests/expected_values/kne_proba_DFP.npy') assert np.allclose(probas, expected)
def test_kne_proba(knn_methods): pool_classifiers, X_dsel, y_dsel, X_test, y_test = setup_classifiers() kne = KNORAE(pool_classifiers, knn_classifier=knn_methods) kne.fit(X_dsel, y_dsel) probas = kne.predict_proba(X_test) expected = np.load( 'deslib/tests/expected_values/kne_proba_integration.npy') assert np.allclose(probas, expected)