DataLearner logo

使用sklearn做高斯混合聚类(Gaussian Mixture Model)

15,156 views

高斯混合模型参考高斯混合模型简介(GMM)

使用sklearn代码如下:

import numpy as np
from sklearn.mixture import GaussianMixture

X = np.loadtxt("d:\\实验\\doc_topic.txt")

gmm = GaussianMixture(n_components=10)
gmm.fit(X)

labels = gmm.predict(X)

print(labels)

DataLearner WeChat

Follow DataLearner WeChat for the latest AI updates

DataLearner 官方微信二维码