toandaominh1997 commited on
Commit
47cb380
·
1 Parent(s): 6426c48
Files changed (1) hide show
  1. app.py +35 -0
app.py ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import pandas as pd
2
+ import streamlit as st
3
+ import vnquant.data as dt
4
+ import seaborn as sns
5
+ import matplotlib.pyplot as plt
6
+ import plotly.express as px
7
+ import plotly.graph_objects as go
8
+ import statsmodels.api as sm
9
+ from statsmodels.tsa.arima.model import ARIMA
10
+ from prophet import Prophet
11
+
12
+
13
+ def prophet_ts(symbol):
14
+ loader = dt.DataLoader(symbol, '2020-01-01','2022-06-01')
15
+ data = loader.download()
16
+ data.columns = [col[0] for col in data.columns]
17
+ m = Prophet()
18
+ pdf = pd.DataFrame()
19
+ pdf['ds'] = data.index
20
+ pdf['y'] = data.close.values
21
+ m.fit(pdf)
22
+ future = m.make_future_dataframe(periods=30)
23
+ forecast = m.predict(future)
24
+ fig = go.Figure()
25
+ fig.add_trace(go.Scatter(x= pdf.ds,
26
+ y=pdf.y,
27
+ name = f"{symbol}_true"
28
+ ))
29
+ fig.add_trace(go.Scatter(x= forecast.ds,
30
+ y=forecast.yhat,
31
+ name = f"{symbol}_pred"
32
+ ))
33
+ return fig
34
+ fig = prophet_ts(symbol='FPT')
35
+ st.plotly_chart(fig, use_container_width=True)