Python监督学习_K近邻分类

本文探讨了Python中的K近邻分类算法。通过Forge数据集展示了当邻居数为1和3时的分类效果,并通过绘制图表来直观理解。接着,介绍了数据集的划分、模型实例化、训练与预测的过程,以及决策边界的可视化。最后,利用cancer数据集分析了模型复杂度与泛化能力之间的关系。
摘要由CSDN通过智能技术生成

首先导入数据集forge(),画出邻居为1的情况的图,横纵轴均为特征值
 

X, y = mglearn.datasets.make_forge()
# mglearn.discrete_scatter(X[:, 0], X[:, 1], y)
# plt.legend(["Class 0", "Class 1"], loc=4)
# plt.xlabel("First feature")
# plt.ylabel("Second feature")
# plt.show()
# print("X.shape:", X.shape)
mglearn.plots.plot_knn_classification(n_neighbors=1)
plt.show()

画出邻居为3的情况

mglearn.plots.plot_knn_classification(n_neighbors=3)
plt.show()

导入数据分好数据集与测试集-->实例化-->分类器进行拟合-->预测-->评估


                
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值