15
правок
Изменения
Подправил код и обновил график в коде
В пакете sklearn алгоритм EM представлен объектом GaussianMixture. Проиллюстрируем его работу на примере задачи кластеризации и сравним его с алгоритмом <tex>k</tex>-means:
[[Файл:Progem_clustering.png|thumb|400px600px|Результат выполнения программы]]
'''import''' numpy as np
'''import''' matplotlib.pyplot as plt
<font color="green"># Для сравнения берем алгоритм - k-means</font>
two_means = cluster.KMeans(n_clusters=n_cluster)
clustering_algorithms = ({ ('GaussianMixtureReal distribution': None, 'Gaussian Mixture': gmm), ('KMeansk-Means', : two_means) )}
for name, algorithm in clustering_algorithms:
# Этап обучения
if algorithm is not None: algorithm.fit(X)
# Применяем алгоритм
# Рисуем результаты
