diff --git a/COVID-19/Prediction/.vs/Prediction/v16/.suo b/COVID-19/Prediction/.vs/Prediction/v16/.suo index 5b02e30..7001ed3 100644 Binary files a/COVID-19/Prediction/.vs/Prediction/v16/.suo and b/COVID-19/Prediction/.vs/Prediction/v16/.suo differ diff --git a/COVID-19/Prediction/Prediction/ARIMA.py b/COVID-19/Prediction/Prediction/ARIMA.py index 6564e8f..d07b6bb 100644 --- a/COVID-19/Prediction/Prediction/ARIMA.py +++ b/COVID-19/Prediction/Prediction/ARIMA.py @@ -8,7 +8,7 @@ import statsmodels.api as sm from sklearn.metrics import mean_squared_error #打开数据文件 -dataset = pd.read_csv('E:\dase intro\COVID-19Analysis\COVID-19\covid-19-all.csv') +dataset = pd.read_csv('E:\DaseIntro\COVID-19Analysis\COVID-19\covid-19-all.csv') #数据预处理 def parse_ymd(s): @@ -24,6 +24,9 @@ CN.index = pd.Index(pd.date_range('2020-01-22','2020-12-09',freq = '1D')) US = dataset[dataset['Country/Region'] == 'US'] US.index = pd.Index(pd.date_range('2020-01-22','2020-12-09',freq = '1D')) +print(CN) +print(US) + #划分训练集、测试集 trainCN = CN[CN['Date'] < '2020-11-01 '] testCN = CN[CN['Date'] >= '2020-11-01'] diff --git a/COVID-19/Prediction/Prediction/ARIMA.pyproj b/COVID-19/Prediction/Prediction/ARIMA.pyproj index 4c5722a..0a538fc 100644 --- a/COVID-19/Prediction/Prediction/ARIMA.pyproj +++ b/COVID-19/Prediction/Prediction/ARIMA.pyproj @@ -24,6 +24,9 @@ + + Code +