add arima

app.py

@@ -66,8 +66,26 @@ class TS:
                                      name = f"{self.symbol}_pred"
                                     ))
         return fig
+    def arima(self, period = 3):
+        df = self.get_data()
+        df.index = df.ds
+        model = ARIMA(df.y, order = (1, 0, 7))
+        model = model.fit()
+        print(len(df))
+        future = pd.DataFrame()
+        predictions = model.forecast(period)
+        future["y"] = predictions
+        print(df.index.max())
+        future.index = pd.date_range(start = df.index.max() + timedelta(days = 1), end = df.index.max() + timedelta(days = period))
+        
+        return self.viz(df, future)
+
 
 
+# ts = TS(symbol = "FPT")
+# data = ts.arima()
+# print(data.index.min(), data.index.max())
+
 
 st.title("Vietnam Trading by Time Series")
 model = st.selectbox("model", ["Prophet", "ARIMA"])
@@ -80,6 +98,6 @@ with st.spinner('Wait for it...'):
     if model == "Prophet":
         fig = ts.prophet(period = period)
     elif model == "ARIMA":
-        fig = ts.prophet(period = period)
+        fig = ts.arima(period = period)
 st.success('Done!')
 st.plotly_chart(fig, use_container_width=True)