机器学习之sklearn-KNN近邻算法分类小案例(乳腺癌预测最优模型)

该博客通过一个乳腺癌预测案例详细介绍了如何利用sklearn库进行KNN分类。首先导入相关库,然后加载数据,划分训练集和测试集。接着进行数据预处理,包括归一化。通过网格搜索找到最优参数,建立KNN模型,并进行模型评估,包括预测准确率、精确率、召回率、F1分数和AUC值。最后,展示了部分测试样本的预测结果和ROC曲线,验证了模型的有效性。
摘要由CSDN通过智能技术生成

(案例):用sklearn包实现knn分类小案例,通过交叉验证网格搜索获取最优参数模型,进行参数评估。
导包:
from sklearn.datasets import load_breast_cancer
from sklearn.model_selection import train_test_split,GridSearchCV
from sklearn.neighbors import KNeighborsClassifier
from sklearn.preprocessing import StandardScaler
from sklearn.metrics import f1_score,precision_score,recall_score,roc_auc_score,roc_curve
import matplotlib.pyplot as plt
import matplotlib as mpl
设置字符集,防止中文乱码
mpl.rcParams[‘font.sans-serif’]=[u’simHei’]
mpl.rcParams[‘axes.unicode_minus’]=False

获取数据:
1.导入乳腺癌数据
data_breast_cancer=load_breast_cancer()
2.获取特征值数据
X=data_breast_cancer.data
3.获取目标值数据
y=data_breast_cancer.target
4.分割数据集,训练集和测试集
X_train,X_test,y_train,y_test=train_test_split(X,y,test_size=0.3,random_

  • 1
    点赞
  • 43
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值