当前位置:   article > 正文

机器学习day2

机器学习day2

一、KNN算法简介

 KNN 算法,或者称 k最邻近算法,是 有监督学习中的分类算法 。它可以用于分类或回归问题,但它通常用作分类算法。

二、KNN分类流程

1.计算未知样本到每一个训练样本的距离

2.将训练样本根据距离大小升序排列

3.取出距离最近的 K 个训练样本

4.进行多数表决,统计 K 个样本中哪个类别的样本个数最多

5.将未知的样本归属到出现次数最多的类别

三、KNN回归流程

1.计算未知样本到每一个训练样本的距离

2.将训练样本根据距离大小升序排列

3.取出距离最近的 K 个训练样本

4.把这个 K 个样本的目标值计算其平均值

5.作为将未知的样本预测的值

四、KNN算法中K的选择

K值过小,过拟合(模型更易受到异常点影响)

K值过大,欠拟合

五、KNN算法API使用

1.分类问题

scikit-learn中,可以使用KNeighborsClassifier类来实现KNN分类。以下是一个简单的示例:

  1. # 导包
  2. from sklearn.neighbors import KNeighborsClassifier
  3. # 导数据
  4. x = [[0, 1, 3], [1, 3, 2], [2, 4, 5], [3, 6, 4]]
  5. y = [0, 0, 1, 1]
  6. # 实例化对象
  7. model = KNeighborsClassifier(n_neighbors=3) #参数n_neighbors指定了在预测时使用的邻居数量
  8. # 训练(-->评估)
  9. model.fit(x, y) #fit计算每个数据点之间的距离
  10. # 预测
  11. myret = model.predict([[4, 3, 7]])
  12. print(f'预测值:{myret}')

2.回归问题

对于回归问题,scikit-learn提供了KNeighborsRegressor类。以下是一个简单的示例:

  1. # 导包
  2. from sklearn.neighbors import KNeighborsRegressor
  3. # 导数据
  4. x = [[1, 2, 3], [2, 4, 5], [2, 3, 6], [4, 5, 1]]
  5. y = [0.1, 0.2, 0.3, 0.4]
  6. # 实例化对象
  7. model = KNeighborsRegressor(n_neighbors=3)
  8. # 训练
  9. model.fit(x, y)
  10. # 评估
  11. mypre = model.predict([[2, 6, 3]])
  12. print(f"预测值:{mypre}")
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/weixin_40725706/article/detail/495139
推荐阅读
相关标签
  

闽ICP备14008679号