第41卷第3期2023年3月水电能源科学WaterResourcesandPowerVol.41No.3Mar.2023DOI:10.20040/j.cnki.1000-7709.2023.20220677基于机器学习算法的万家寨水库排沙预测研究颜小飞1,2,郭秀吉1,2,孙龙飞1,2(1.黄河水利委员会黄河水利科学研究院,河南郑州450003;2.水利部黄河下游河道与河口治理重点实验室,河南郑州450003)摘要:为克服水库排沙多因素、非线性复杂关系建立难题,实现水库排沙准确预测,利用万家寨水库2002~2020年水沙系列数据,基于XGBoost、KNN、SVR、RF四种机器学习算法分别预测分析水库出库含沙量。结果表明,利用机器学习算法可有效预测综合考虑不同影响因素的水库排沙;不同机器学习算法在水库排沙预测的适用性有所不同,对比之下,基于RF算法建立的水库排沙预测模型的决定系数R2最高为0.9349,平均绝对误差及均方根误差均最小,分别为2.974、4.886,其预测效果更优于其他三种算法。研究成果可为水库排沙精确预测及调度方案优化提供参考。关键词:水库排沙;含沙量;机器学习算法;预测模型中图分类号:TV145文献标志码:A文章编号:1000-7709(2023)03-0079-04收稿日期:2022-04-07,修回日期:2022-06-13基金项目:国家重点研发计划(2021YFC3200400);黄河水利科学研究院科技发展基金专项项目(黄科发202102);黄河水利科学研究院基本科研业务费专项(HKY-JBYW-2018-18)作者简介:颜小飞(1985-),男,硕士、工程师,研究方向为测绘物探仪器应用,E-mail:yxf513@126.com通讯作者:孙龙飞(1993-),男,博士、工程师,研究方向为水工结构数值模拟与施工质量实时控制,E-mail:1287215017@qq.com1概况黄河万家寨水利枢纽位于黄河中游上段托克托至龙口峡谷河段内(图1)。万家寨水库于1998年10月下闸蓄水,水库总库容为8.96×108m3,调节库容为4.45×108m3,死库容为4.51×108m3,设计调洪库容为3.02×108m3;最高蓄水位为980.00m,正常蓄水位为977.00m,校核洪水位为979.10m,汛期限制水位为966.00m,最低发电水位为952...