Data-driven Modeling and Prediction of COVID-19 Infection