def test_kmp_create_components_kmeans_stratified(): components = create_components(mult_dense, mult_target, n_components=0.25, class_distrib="stratified", random_state=0) assert_equal(components.shape, (mult_dense.shape[0]/4 - 1, mult_dense.shape[1]))
def create_kmeans_comp(X_train, y_train, class_distrib, n_components, random_state): return create_components(X_train, y_train, n_components=n_components, class_distrib=class_distrib, random_state=random_state, verbose=1)
def test_kmp_create_components_kmeans_global(): components = create_components(mult_dense, mult_target, n_components=0.25, random_state=0) assert_equal(components.shape[0], mult_dense.shape[0]/4)