def test_des_clustering(): pool_classifiers, X_dsel, y_dsel, X_test, y_test = setup_classifiers() rng = np.random.RandomState(123456) des_clustering = DESClustering(pool_classifiers, rng=rng) des_clustering.fit(X_dsel, y_dsel) assert np.isclose(des_clustering.score(X_test, y_test), 0.97872340425531912)
def test_des_clustering(): from sklearn.cluster import KMeans pool_classifiers, X_dsel, y_dsel, X_test, y_test = setup_classifiers() rng = np.random.RandomState(123456) cluster = KMeans(n_clusters=5, random_state=rng) des_clustering = DESClustering(pool_classifiers, clustering=cluster) des_clustering.fit(X_dsel, y_dsel) assert np.isclose(des_clustering.score(X_test, y_test), 0.97872340425531912)