def test_desknn(): pool_classifiers, X_dsel, y_dsel, X_test, y_test = setup_classifiers() desknn = DESKNN(pool_classifiers, DFP=True) desknn.fit(X_dsel, y_dsel) assert np.isclose(desknn.score(X_test, y_test), 0.89393939393939392)
def test_desknn(knn_methods): pool_classifiers, X_dsel, y_dsel, X_test, y_test = setup_classifiers() desknn = DESKNN(pool_classifiers, knn_classifier=knn_methods) desknn.fit(X_dsel, y_dsel) assert np.isclose(desknn.score(X_test, y_test), 0.97340425531914898)