Add init

README.md

@@ -1,13 +1,14 @@
 ---
-title: Trader
-emoji: 🌖
-colorFrom: yellow
-colorTo: gray
+title: trader
+emoji: ⚡
+colorFrom: purple
+colorTo: yellow
 sdk: streamlit
-sdk_version: 1.10.0
+sdk_version: 1.2.0
 app_file: app.py
 pinned: false
 license: apache-2.0
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces#reference
+

app.py

@@ -0,0 +1,35 @@
+import pandas as pd
+import streamlit as st
+import vnquant.data as dt
+import seaborn as sns
+import matplotlib.pyplot as plt
+import plotly.express as px
+import plotly.graph_objects as go
+import statsmodels.api as sm
+from statsmodels.tsa.arima.model import ARIMA
+from prophet import Prophet
+
+
+def prophet_ts(symbol):
+    loader = dt.DataLoader(symbol, '2020-01-01','2022-06-01')
+    data = loader.download()
+    data.columns = [col[0] for col in data.columns]
+    m = Prophet()
+    pdf = pd.DataFrame()
+    pdf['ds'] = data.index 
+    pdf['y'] = data.close.values
+    m.fit(pdf)
+    future = m.make_future_dataframe(periods=30)
+    forecast = m.predict(future)
+    fig = go.Figure()
+    fig.add_trace(go.Scatter(x= pdf.ds, 
+                            y=pdf.y,
+                            name = f"{symbol}_true"
+                            ))
+    fig.add_trace(go.Scatter(x= forecast.ds, 
+                                 y=forecast.yhat,
+                                 name = f"{symbol}_pred"
+                                ))
+    return fig
+fig = prophet_ts(symbol='FPT')
+st.plotly_chart(fig, use_container_width=True)

requirements.txt

@@ -0,0 +1,7 @@
+scikit-learn
+seaborn
+plotly
+statsmodels
+prophet
+"git+https://github.com/phamdinhkhanh/vnquant.git@master"
+