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)
