kmeans監督學習算法示例
命令語句:用sklearn的任何一個監督模型包的訓練語句和預測語句。
# skleanrn 語句
from sklearn.cluster import KMeans
import numpy as np
X = np.array([[1, 2], [1, 4], [1, 0],
[4, 2], [4, 4], [4, 0]])
kmeans = KMeans(n_clusters=2, random_state=0).fit(X)
kmeans.labels_
np.array([0, 0, 0, 1, 1, 1], dtype=np.int32)
kmeans.predict([[0, 0], [4, 4]])