基于机器学习的水质可饮用性预测

用水质理化指标快速预测水是否可饮用:基于 WHO 标准做领域特征工程 + SMOTE + 阈值优化 + 熵权-TOPSIS 客观选模 + SHAP 可解释——带注释代码、技术文档、面试问答全配齐。

  • 任务类型机器学习
  • 专业方向计算机 · 环境/地理 · 数据科学

项目亮点

  • 二分类任务:水质可饮用性判断
  • 数据不平衡处理:SMOTE过采样
  • 特征工程:pH偏离度、硬度/固体比、氯胺/有机碳比
  • 两阶段调参:贝叶斯粗调 + 网格细调

数据与任务

样本量3276 样本 · 9 理化指标
核心方法6 模型 + TOPSIS
技术栈scikit-learn · SHAP

如果你想找一个贴近公共卫生、又把不平衡分类方法论做全的项目,这个「水质可饮用性预测」很合适。

它是一个完整的二分类工程案例,配套也给你备齐了,帮你真正搞懂它、在面试和答辩里讲明白:带中文注释、能读懂的代码,一份从背景讲到每步实现的技术文档,一份把面试问题连答案都写好的问答文档,还有一整套能直接做 PPT 的配图。

flowchart LR A["水质理化指标"] --> B["缺失值处理<br/>+ WHO 特征工程"] B --> C["SMOTE 平衡"] C --> D["六模型 + 三重调参"] D --> E["熵权 + TOPSIS 选模"] D --> F["SHAP 可解释"] E --> G["可饮用 / 不可饮用"] F --> G

先说清楚,它到底在做什么

判断一份水能不能喝,传统要靠实验室化学分析,耗时又贵,没法实时监测。难点在于:水质数据的两类样本在特征空间高度重叠、还存在缺失和类别不平衡,是个相当有挑战的分类问题。

这个项目用机器学习从理化指标(pH、硬度、氯胺、浊度……)快速预测水质是否达到 WHO 饮用标准。关键在一套扎实的方法论:基于 WHO 标准构造领域特征(如 pH 偏离标准的程度)→ 用 SMOTE 平衡类别 → 三重调参(贝叶斯 + 网格 + 阈值优化)→ 熵权-TOPSIS 客观选模 → SHAP 可解释。

WHO 标准对比
项目把各项指标和 WHO 饮用标准做了对比——这是构造领域特征的依据。

搞懂它,你能在面试里讲清楚什么

把下面几件事吃透,面试官顺着问下来你都能接得住。

基于 WHO 标准的领域特征工程。 不直接用原始指标,而是构造"偏离 WHO 标准的程度"这类特征,把领域知识显式喂给模型,让它更容易学到评估逻辑。你能借此讲清楚领域特征工程的价值。

SMOTE + 阈值优化这套组合拳。 SMOTE 在少数类的近邻间合成新样本来平衡训练集;平衡后默认 0.5 的判定阈值往往不再合适,所以再遍历阈值找最优 F1。你能讲清楚"为什么平衡之后还要调阈值",这是不平衡分类的关键细节。

SMOTE 与阈值优化
照着这张图,能把"SMOTE 平衡 + 阈值优化"为什么要配合使用讲明白。

熵权-TOPSIS 客观选模 + SHAP 可解释。 多个指标各有侧重,熵权法按区分度自动算权重、TOPSIS 综合排名客观选模;SHAP 再把每个理化指标的影响方向画出来。

SHAP 特征影响
SHAP 蜂群图:哪些理化指标在影响判定,一目了然,面试讲这个很加分。

下面这组分析图也都给你做好了,可以直接放进答辩或面试 PPT

ROC 对比
多模型 ROC 对比
阈值曲线
阈值-F1 优化曲线

更关键的是,每张图怎么跑出来的、该怎么解读,技术文档里都讲清楚了——你能说明白每张图到底说明了什么。

面试官会问的,都帮你备好了

随便感受几个这个项目真实会被追问的问题:

  • SMOTE 为什么只在训练集上做,而不是整个数据集?
  • 平衡之后为什么最优阈值不再是 0.5?阈值该怎么定?
  • 熵权-TOPSIS 综合评价比只看单一指标好在哪?

看到会愣一下?正常。配套的面试问答文档把这个项目——从整体思路到每个流程细节、各种可能被追问的点——连参考答案都给你写好了。另外还有现成的简历描述,照着改就能写进简历;那套配图也能直接套进 PPT 模板,快速出一份面试 / 答辩 PPT。

配套资料:搞懂一个项目需要的,这里全都有

先看那份技术文档——从背景、特征工程、不平衡处理一直讲到 TOPSIS 与 SHAP,图文并茂:

文档封面
封面 + 目录
方法页
不平衡处理与调参
结果页
评估与可解释性

代码也给你了——关键部分都带着中文注释,帮你读懂"它到底是怎么实现的"

特征工程代码
WHO 标准相关特征工程
TOPSIS 代码
熵权 + TOPSIS 综合排名

技术文档、面试问答、源码注释、整套配图——搞懂一个项目、并在面试里讲清楚它,需要的全都备齐了。

适合谁

不管你是赶毕设、想给简历添个有分量的项目,还是在准备面试,这个题目都接得住。专业上,计算机、数据科学、环境工程、公共卫生、人工智能方向都很合适。不平衡分类是工业界最常见的难题之一,把这套方法论真正搞懂、能讲出来,就是一个能写进简历、撑得起面试的项目。

想把这样的项目做成你简历上的亮点?

这是一套配齐了代码、文档、面试问答和配图的 AI+X 项目,可写进简历、在面试里讲清楚。 想做同类项目、或获取「基于机器学习的水质可饮用性预测」的完整资料(代码 / 数据处理流程 / 论文文档 / 配图), 请联系为你介绍本页面的老师咨询,按你的情况定一个合适的项目。