diff --git a/tensor-eval.py b/tensor-eval.py index a4cd9a3..873f39d 100644 --- a/tensor-eval.py +++ b/tensor-eval.py @@ -7,7 +7,7 @@ from sklearn.metrics import mean_squared_error as rmse reg = LinearRegression() alldata = pd.read_csv( - 'country_vaccinations.csv', header=0, skipinitialspace=True, + 'test.csv', header=0, skipinitialspace=True, usecols=['total_vaccinations', 'people_vaccinated', 'daily_vaccinations' ,'daily_vaccinations_per_million']).dropna() X = alldata[[c for c in alldata.columns if c != 'daily_vaccinations']]