def train_model_on_country(prophet_df, country="US"):
# Train model
m = Prophet( #daily_seasonality = True,
#yearly_seasonality = False,
#weekly_seasonality = True,
#growth='linear',
interval_width=0.68 # one sigma
)
m.add_country_holidays(country_name=country)
m.fit(prophet_df)
return m
