tonne commited on
Commit
965165e
1 Parent(s): 4f06acc

change vnquant to vnstock

Browse files
Files changed (2) hide show
  1. app.py +7 -7
  2. requirements.txt +1 -1
app.py CHANGED
@@ -1,6 +1,6 @@
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
@@ -17,9 +17,9 @@ start_date = str((datetime.now(pytz.timezone('Asia/Ho_Chi_Minh')) - timedelta(da
17
  end_date = str((datetime.now(pytz.timezone('Asia/Ho_Chi_Minh')) - timedelta(days=0)).strftime("%Y-%m-%d"))
18
 
19
  def prophet_ts(symbol, periods = 10):
20
- loader = dt.DataLoader(symbol, start_date, end_date)
21
- data = loader.download()
22
- data.columns = [col[0] for col in data.columns]
23
  m = Prophet()
24
  pdf = pd.DataFrame()
25
  pdf['ds'] = data.index
@@ -41,9 +41,9 @@ class TS:
41
  def __init__(self, symbol):
42
  self.symbol = symbol
43
  def get_data(self):
44
- loader = dt.DataLoader(self.symbol, start_date, end_date)
45
- data = loader.download()
46
- data.columns = [col[0] for col in data.columns]
47
  pdf = pd.DataFrame()
48
  pdf['ds'] = data.index
49
  pdf['y'] = data.close.values
 
1
  import pandas as pd
2
  import streamlit as st
3
+ from vnstock import *
4
  import seaborn as sns
5
  import matplotlib.pyplot as plt
6
  import plotly.express as px
 
17
  end_date = str((datetime.now(pytz.timezone('Asia/Ho_Chi_Minh')) - timedelta(days=0)).strftime("%Y-%m-%d"))
18
 
19
  def prophet_ts(symbol, periods = 10):
20
+ data = stock_historical_data(symbol=symbol,
21
+ start_date=start_date,
22
+ end_date=end_date, resolution='1D', type='stock')
23
  m = Prophet()
24
  pdf = pd.DataFrame()
25
  pdf['ds'] = data.index
 
41
  def __init__(self, symbol):
42
  self.symbol = symbol
43
  def get_data(self):
44
+ data = stock_historical_data(symbol=self.symbol,
45
+ start_date=start_date,
46
+ end_date=end_date, resolution='1D', type='stock')
47
  pdf = pd.DataFrame()
48
  pdf['ds'] = data.index
49
  pdf['y'] = data.close.values
requirements.txt CHANGED
@@ -3,7 +3,7 @@ seaborn
3
  plotly
4
  statsmodels
5
  prophet
6
- git+https://github.com/phamdinhkhanh/vnquant.git@master
7
  httpx
8
  pytz
9
 
 
3
  plotly
4
  statsmodels
5
  prophet
6
+ vnstock
7
  httpx
8
  pytz
9