The Prediction of Hepatitis E through Ensemble Learning.

Journal: International journal of environmental research and public health
Published Date:

Abstract

According to the World Health Organization, about 20 million people are infected with Hepatitis E every year. In 2015, there were 44,000 deaths due to HEV infection worldwide. Food, water and climate are key factors that affect the outbreak of Hepatitis E. This paper presents an ensemble learning model for Hepatitis E prediction by studying the correlation between historical epidemic cases of hepatitis E and environmental factors (water quality and meteorological data). Environmental factors include many features, and ones that are most relevant to HEV are selected and input into the ensemble learning model composed by Gradient Boosting Decision Tree (GBDT) and Random Forest for training and prediction. Three indicators, root mean square error (RMSE), mean absolute error (MAE) and mean absolute percentage error (MAPE), are used to evaluate the effectiveness of the ensemble learning model against the classical time series prediction model. It is concluded that the ensemble learning model has a better prediction effect than the classical model, and the prediction effectiveness can be improved by exploiting water quality and meteorological factors (radiation, air pressure, precipitation).

Authors

  • Tu Peng
    School of Computer Science and Technology, Beijing Institute of Technology, Beijing 100081, China.
  • Xiaoya Chen
    Shanghai Mental Health Center, Shanghai Jiao Tong University, School of Medicine, Shanghai, China.
  • Ming Wan
    Chinese Center for Disease Control and Prevention, 102206 Beijing, China.
  • Lizhu Jin
    Chinese Center for Disease Control and Prevention, 102206 Beijing, China.
  • XiaoFeng Wang
    Indiana University Bloomington.
  • Xuejie Du
    Chinese Center for Disease Control and Prevention, 102206 Beijing, China.
  • Hui Ge
    Chinese Center for Disease Control and Prevention, 102206 Beijing, China.
  • Xu Yang
    Department of Food Science and Technology, The Ohio State University, Columbus, OH, United States.