栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

KNN(K相邻总结)

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

KNN(K相邻总结)

KNN(K-NearestNeighbor,又称K相邻)算法是机器学习最简单的方法之一。
特性:
1.思想极度简单,应用数学知识少(近乎为零),对于很多不擅长数学的小伙伴十分友好
2.适用于样本容量比较大的类域的自动分类
算法流程:
①准备数据(包括清洗、数据集划分)。
②计算测试样本点(也就是待分类点)到其他每个样本点的距离 。
③对每个距离进行排序,然后选择出距离最小的K个点。
④对K个点所属的类别进行比较,根据少数服从多数的原则,将测试样本点归入在K个点中占比最高的那一类。
scikit-learn的KNN API
以自带鸢尾花为例子(不涉及清洗以及集合划分):

from sklearn.datasets import load_iris
from sklearn.preprocessing import StandardScaler
from sklearn.neighbors import KNeighborsClassifier

if __name__ == '__main__':
    # 1. 加载数据集  
    iris = load_iris() #通过iris.data 获取数据集中的特征值  iris.target获取目标值

    # 2. 数据标准化
    transformer = StandardScaler()
    x_ = transformer.fit_transform(iris.data) # iris.data 数据的特征值

    # 3. 模型训练
    estimator = KNeighborsClassifier(n_neighbors=3) # n_neighbors 邻居的数量,也就是Knn中的K值
    estimator.fit(x_, iris.target) # 调用fit方法 传入特征和目标进行模型训练

    # 4. 利用模型预测
    result = estimator.predict(x_) 
    print(result)

K值如何取最优?sklearn提供了网格搜索的方法(常用param_grid需要传递的K值列表;cv=None交叉验证参数,默认None;verbose:日志冗长度,int:冗长度,0:不输出训练过程,1:偶尔输出,>1:对每个子模型都输出。)代码如下:

import numpy as np
# 加载莺尾花数据集
from sklearn import datasets
# 导入KNN分类器
from sklearn.neighbors import KNeighborsClassifier
from sklearn.model_selection import train_test_split
from sklearn.model_selection import GridSearchCV
from sklearn.preprocessing import StandardScaler  # 标准化
from sklearn.preprocessing import MinMaxScaler   # 归一化

isir = datasets.load_iris()
x,y = isir.data,isir.target
X_train, X_test, y_train, y_test = train_test_split(x,y,test_size=0.2,random_state=0)
transform = StandardScaler()
X_train = transform.fit_transform(X_train)
# X_test = transform.fit(X_test)
X_test = transform.fit_transform(X_test)
n_neighbors_list = {'n_neighbors': [3,4,5,6,7]}
estimator = KNeighborsClassifier()
estimator = GridSearchCV(estimator,param_grid=n_neighbors_list,verbose=0,cv=5)
estimator.fit(X_train,y_train)
# 打印最优参数
print('最优参数组合:', estimator.best_params_, '最好得分:', estimator.best_score_)
# 测试集评估模型
print('测试集准确率:', estimator.score(X_test, y_test))
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/840020.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号