prophet forecast model

prophet_model.py

def run_prophet(series):
  model = Prophet(daily_seasonality=False,
                  weekly_seasonality=False,
                  yearly_seasonality=False)
  model.fit(series)
  forecast = model.make_future_dataframe(periods=10, freq='Y')
  forecast = model.predict(forecast)
  forecast = forecast[['ds', 'yhat', 'yhat_lower', 'yhat_upper']]
  forecast['City,State'] = series['City,State'].iloc[0]
  forecast = forecast[['City,State','ds', 'yhat', 'yhat_lower', 'yhat_upper']]
  forecast['ds'] = pd.DatetimeIndex(forecast['ds']).year
  forecast[['yhat', 'yhat_lower', 'yhat_upper']] = forecast[['yhat', 'yhat_lower', 'yhat_upper']].astype(float)
  forecast[['yhat', 'yhat_lower', 'yhat_upper']] = forecast[['yhat', 'yhat_lower', 'yhat_upper']].apply(np.ceil)
  return forecast
  
for i in range(len(series)):
  f = run_prophet(series[i])
  f.to_csv('csv/population_prediction.csv', mode='a', index='False')