def locaComparison(location1, location2):
ln = LinearRegression()
ln.init_data(location1, location2)
ln.compare('cases', location1, location2)
ln.compare('deaths', location1, location2)
def plotGraph(country):
ln = LinearRegression()
ln.init_data(country)
ln.predict('New cases of ' + country, 'poly')
ln.predict('New deaths of ' + country, 'poly')
