Expectation maximization algorithm
YAML configuration
expectationmaximization: name: Expectation Maximization algo: GaussianMixture description: Similar to K-Means, but doesn't assign hard clusters to points n_components: 4 covariance_type: full
geocluster.py
elif algo.lower() == 'gaussianmixture': cluster_cls = mixture.GaussianMixture
and calculating labels
else: model = clusterer.fit(cluster_data) if algo.lower() != 'gaussianmixture': labels = model.labels_ else: labels = model.predict(cluster_data)
This configuration will help setup EM clustering algorithm.