一种基于机器学习的湖泊水体重金属浓度预测的方法

本发明公开一种基于机器学习的湖泊水体重金属浓度预测方法，通过采集湖泊水体理化性质并建立湖泊数据库；标准化处理特征数据集；基于无监督机器学习将特征数据集输入到鲁棒主成分分析(RPCA)和自组织映射(SOM)进行降维，获得含有RPCA特征和SOM特征的新特征数据集；将新特征数据集应用到机器学习支持向量机模型、随机森林模型和多层感知机模型进行训练，得到可预测湖泊水体重金属浓度的机器学习模型；依据训练好的机器学习预测模型的10折交叉验证结果，最后选取预测结果最好的随机森林模型进行预测，得到湖泊水体中重金属的浓度。本发明便捷、准确的预测湖泊水体重金属浓度的性质，为快速判断湖泊及其所在区域的污染程度提供了技术基础。