import numpy as np
from sklearn import datasets
from sklearn.cluster import KMeans
X, y = datasets.load_iris(return_X_y=True)
nSample, nDim = X.shape
model = KMeans(n_clusters=3)
y_hat = model.fit_predict(X)
labs = np.unique(y_hat)
means = np.zeros((len(labs),nDim))
for i, lab in enumerate(labs):
means[i] = np.mean(X[y_hat == lab], axis=0)
print(means)
km = KMeans(n_clusters=3).fit(X)
print(km.cluster_centers_)
获取K-means的聚类中心
于 2022-01-22 19:50:59 首次发布