The main function is testing.R, you can call sarimax_prediction_result <- predict_sarimax(data, window_size) to get the same result table of real cases and predicting cases, the same as the no-cov.