目录
一、KNN
-
- 一、KNN
- 二、核心代码
- 三、评价指标
-
- 3.1 混淆矩阵
- 3.2 预测分数
- 3.3 召回率
- 3.4 F分数
- 3.5 FN(false negative)
- 3.6 ROC曲线
- 3.7 AUC指标
KNN模型步骤:
- 加载数据集
- 初始化k的值
- 为了获得预测的类,从1迭代到训练数据点的总数
- 计算测量数据和每行训练数据之间的距离(最常用的为欧几里得距离);
- 根据距离值按升序对计算的距离进行排序
- 从排序的数组中获取前k行
- 获取这些行中频率最高的类
- 返回预测的类
首先需要导入相应库和数据集:
import pandas