#Model Testing Code =.py

#Model Testing Code = 

from sklearn.model_selection import train_test_split
from sklearn.linear_model import LinearRegression
from sklearn.metrics import mean_squared_error, r2_score

#This splitts the data
X = df[['Population Staying at Home', 'Week']]
y = df['Percent Change in Visits']
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)


model = LinearRegression()
model.fit(X_train, y_train)

#This is responsible for predicting the data 
y_pred_train = model.predict(X_train)
y_pred_test = model.predict(X_test)