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