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)