通过sklearn中的make_blobs函数生成数据集,进行聚类操作,任选另一种聚类算法完成聚类,并进行可视化
时间: 2024-05-08 13:20:10 浏览: 99
好的,我理解了你的问题。你可以使用KMeans算法代替make_blobs函数生成的数据集进行聚类操作。首先,导入必要的库:
```python
import numpy as np
import matplotlib.pyplot as plt
from sklearn.datasets import make_blobs
from sklearn.cluster import KMeans
```
然后,使用make_blobs函数生成数据集:
```python
X, y = make_blobs(n_samples=300, centers=4, cluster_std=0.60, random_state=0)
```
该函数生成300个样本,4个中心,标准差为0.6的数据集。下一步是使用KMeans聚类算法来对数据进行聚类,代码如下:
```python
kmeans = KMeans(n_clusters=4)
kmeans.fit(X)
```
接下来,我们将绘制聚类结果图。代码如下:
```python
plt.scatter(X[:, 0], X[:, 1], c=kmeans.labels_, cmap='viridis')
centers = kmeans.cluster_centers_
plt.scatter(centers[:, 0], centers[:, 1], c='black', s=200, alpha=0.5);
```
该代码使用散点图显示数据集,并用不同颜色表示不同的聚类,中心点用黑色实心圆表示。这样就可以可视化聚类结果了。
阅读全文