Skip to content

KNN K 近邻分类

本章目标

  1. 明确本分册对应的 KNN 源码入口与运行方式。
  2. 理解当前 KNN 文档各章节分别负责解释什么内容。
  3. 建立从数据、模型、训练到评估的整体阅读路线。

对应代码速览

组件路径说明
数据生成data_generation/classification.pyClassificationData.knn() 生成双月牙二分类数据
数据导出data_generation/__init__.py导出 knn_data
训练封装model_training/classification/knn.pytrain_model(...) 封装 sklearn.neighbors.KNeighborsClassifier 训练
端到端流水线pipelines/classification/knn.py完成切分、标准化、训练、预测与可视化
混淆矩阵可视化result_visualization/confusion_matrix.py绘制预测结果混淆矩阵
ROC 曲线可视化result_visualization/roc_curve.py绘制二分类 ROC 曲线
决策边界可视化result_visualization/decision_boundary.py绘制 PCA 2D 空间下的决策边界
学习曲线可视化result_visualization/learning_curve.py绘制训练/验证得分曲线

默认配置速览(来自源码)

项目当前实现
训练模型KNeighborsClassifier(n_neighbors=5, weights='uniform', metric='minkowski')
数据切分train_test_split(..., test_size=0.2, random_state=42, stratify=y)
特征预处理StandardScaler 仅在训练集 fit,测试集 transform——KNN 依赖距离度量,标准化是必需的
正式预测输出y_pred = model.predict(X_test_s)
概率输出y_scores = model.predict_proba(X_test_s)(当前源码做接口存在性检查)
评估方式混淆矩阵 + ROC 曲线 + PCA 2D 决策边界 + 学习曲线

阅读路线

  1. 数学原理
  2. 数据构成
  3. 思路与直觉
  4. 模型构建
  5. 训练与预测
  6. 评估与诊断
  7. 工程实现
  8. 练习与参考文献

如何运行

示例代码

bash
python -m pipelines.classification.knn

理解重点

  • 这个命令会串起当前 KNN 分册中最核心的工程流程。
  • 运行后会训练一个 KNN 模型(懒惰学习,本质是存储训练样本),并输出混淆矩阵、ROC 曲线、决策边界图和学习曲线。
  • 当前任务是监督二分类,因此 label 会真实参与模型拟合与测试集预测。

先修

小结

  • 本分册严格对应当前仓库中的 KNN 源码实现。
  • KNN 与其他分类算法(逻辑回归、SVC、决策树)的核心区别在于:它是懒惰学习,不对数据做显式参数拟合,预测时才计算邻域关系。
  • 如果已经熟悉整体入口,可以直接从"数据构成""模型构建"或"训练与预测"章节开始阅读。