Merge pull request #153 from bfxavier/fixing-prophet

Fixing prophet run
This commit is contained in:
Artem Veremey
2021-03-11 12:27:10 -08:00
committed by GitHub

View File

@@ -44,8 +44,8 @@ def fbprophet(l_args, s_ticker, df_stock):
df_stock = df_stock.sort_index(ascending=True)
df_stock.reset_index(level=0, inplace=True)
df_stock = df_stock[["date", "5. adjusted close"]]
df_stock = df_stock.rename(columns={"date": "ds", "5. adjusted close": "y"})
df_stock = df_stock[["Date", "5. adjusted close"]]
df_stock = df_stock.rename(columns={"Date": "ds", "5. adjusted close": "y"})
df_stock["ds"] = pd.to_datetime(df_stock["ds"])
model = Prophet(yearly_seasonality=False, daily_seasonality=False)