add arima

app.py

@@ -71,10 +71,9 @@ class TS:
         df.index = df.ds
         model = ARIMA(df.y, order = (1, 0, 7))
         model = model.fit()
-        print(len(df))
         forecast = pd.DataFrame()
         predictions = model.forecast(period)
-        forecast["y"] = predictions
+        forecast["yhat"] = predictions
         forecast["ds"] = pd.date_range(start = df.index.max() + timedelta(days = 1), end = df.index.max() + timedelta(days = period))
         
         return self.viz(df, forecast)