Python生成已分类数据集

最新推荐文章于 2024-10-02 17:22:34 发布

小白拉普拉斯

最新推荐文章于 2024-10-02 17:22:34 发布

阅读量2.1k

点赞数 1

分类专栏：机器学习文章标签： KNN(K近邻)算法

本文链接：https://blog.csdn.net/XiaobaiLaplace/article/details/95100291

版权

本文介绍了如何使用scikit-learn库的make_blobs方法生成聚类算法的测试数据，这些数据适用于评估K近邻（KNN）算法的效果。通过指定特征数量、中心点和数据范围，可以创建多类别数据集。

摘要由CSDN通过智能技术生成

1. 代码

from sklearn.datasets import make_blobs#使用make_blobs进行knn分类
from sklearn.neighbors import KNeighborsClassifier#导入KNN分类器
import matplotlib.pyplot as plt#导入画图
from sklearn.model_selection import train_test_split#导入数据集拆分工具
data=make_blobs(n_samples=200,centers=2,random_state=8)
X,y=data
plt.scatter(X[:,0],X[:,1],c=y,cmap=plt.cm.spring,edgecolor='k')
plt.show()