长江科学院院报 ›› 2020, Vol. 37 ›› Issue (5): 43-49.DOI: 10.11988/ckyyb.20181326

• 水资源与环境 • 上一篇    下一篇

基于多粒度特征和XGBoost模型的城市日供水量预测

贺波1, 马静1, 高赫余2   

  1. 1.南京航空航天大学 经济与管理学院,南京 210016;
    2.上海城投水务(集团)有限公司 供水分公司,上海 200001
  • 收稿日期:2018-12-17 出版日期:2020-05-01 发布日期:2020-06-10
  • 通讯作者: 马 静(1966-),女,四川成都人,教授,博士,主要从事大数据分析方面的研究。E-mail:majing5525@126.com
  • 作者简介:贺 波(1995-),男,陕西榆林人,硕士,主要从事数据挖掘方面的研究。E-mail:18215609986@163.com
  • 基金资助:
    国家自然科学基金面上项目 (71373123);中央高校基本科研业务费专项(NW2018004)

Predicting Urban Daily Water Supply Based on Multi-granularityFeature and XGBoost Integrated Model

HE Bo1, MA Jing1, GAO He-yu2   

  1. 1. School of Economics and Management, Nanjing University of Aeronautics and Astronautics,Nanjing 210016, China;
    2. Shanghai Chengtou Water Company, Shanghai 200001, China
  • Received:2018-12-17 Published:2020-05-01 Online:2020-06-10

摘要: 城市日供水量预测对供水部门具有十分重要的现实意义。为提高城市日供水量预测精度,以某市历史用水数据为基础,构建多粒度特征,并利用Pearson相关系数进行特征的筛选,基于XGBoost模型构建城市日供水量预测模型。本模型通过在训练集上进行训练和学习,在测试集上的平均绝对误差为70 571 t/d,平均相对误差为1.4%;传统的回归预测方法如随机森林法和支持向量机法,平均绝对误差分别为84 366 t/d和88 848 t/d。本模型预测精度更高,说明此模型可行、有效,具有一定的应用价值。

关键词: 城市日供水量, 多粒度特征, Pearson相关系数, XGBoost模型, 预测精度

Abstract: Predicting the quantity of urban daily water supply is of great significance to water supply department in practice. To ameliorate the accuracy of predicting urban water supply, an XGBoost (eXtreme Gradient Boosting) integrated model of predicting the urban water supply is built based on the historical data of water supply with multi-granular features. Pearson correlation coefficient is used to select the optimal factor combination. Through training and learning on the training set, the results show that the average absolute error of this model is 70 571 t/d, and the average relative error is 1.4% on test set. Compared with traditional regression prediction methods such as random forest and support vector machine with the average absolute error amounting to 84 366 t/d and 88 848 t/d, respectively, the present method has higher prediction accuracy, indicating that the model is feasible and effective.

Key words: urban daily water supply, multi-granularity feature, Pearson correlation coefficient, XGBoost model, prediction accuracy

中图分类号: