KNN K 近邻分类
本章目标
- 明确本分册对应的 KNN 源码入口与运行方式。
- 理解当前 KNN 文档各章节分别负责解释什么内容。
- 建立从数据、模型、训练到评估的整体阅读路线。
对应代码速览
| 组件 | 路径 | 说明 |
|---|---|---|
| 数据生成 | data_generation/classification.py | ClassificationData.knn() 生成双月牙二分类数据 |
| 数据导出 | data_generation/__init__.py | 导出 knn_data |
| 训练封装 | model_training/classification/knn.py | train_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 决策边界 + 学习曲线 |
阅读路线
如何运行
示例代码
bash
python -m pipelines.classification.knn理解重点
- 这个命令会串起当前 KNN 分册中最核心的工程流程。
- 运行后会训练一个 KNN 模型(懒惰学习,本质是存储训练样本),并输出混淆矩阵、ROC 曲线、决策边界图和学习曲线。
- 当前任务是监督二分类,因此
label会真实参与模型拟合与测试集预测。
先修
小结
- 本分册严格对应当前仓库中的 KNN 源码实现。
- KNN 与其他分类算法(逻辑回归、SVC、决策树)的核心区别在于:它是懒惰学习,不对数据做显式参数拟合,预测时才计算邻域关系。
- 如果已经熟悉整体入口,可以直接从"数据构成""模型构建"或"训练与预测"章节开始阅读。