Day19 K平均演算法实作

https://github.com/PacktPublishing/Machine-Learning-Algorithms

使用sci-kit learn

一样先导入套件，上面的是用来算数学的;下面的是用来画画的，并且帮它们取绰号(np & plt)。

import numpy as np
import matplotlib.pyplot as plt

再来，用seed()随机产生整数的乱数後，设定资料集的数量100，n_features=2、centers=3代表我们希望产生三个二维的群集，而且不会储存Y(_取代)，因为Y的值是一维的。

from sklearn.datasets import make_blobs
np.random.seed(1000)
nb_samples = 1000
X, _ = make_blobs(n_samples=nb_samples, n_features=2, centers=3, cluster_std=1.5)

可以看到以下是我们的结果，资料被分成三群了。

from sklearn.cluster import KMeans
km = KMeans(n_clusters=3)
km.fit(X)