From 29b0d829ddf7ebf7d3e9d2874338d8ea8848026f Mon Sep 17 00:00:00 2001 From: Bruno Xavier Date: Thu, 11 Mar 2021 05:48:35 -0800 Subject: [PATCH] fixing prophet run --- gamestonk_terminal/prediction_techniques/fbprophet.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/gamestonk_terminal/prediction_techniques/fbprophet.py b/gamestonk_terminal/prediction_techniques/fbprophet.py index a752813870a..51336bd6f63 100644 --- a/gamestonk_terminal/prediction_techniques/fbprophet.py +++ b/gamestonk_terminal/prediction_techniques/fbprophet.py @@ -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)