tonne commited on
Commit
59b3004
1 Parent(s): 9830929

change vnquant to vnstock

Browse files
Files changed (1) hide show
  1. app.py +14 -6
app.py CHANGED
@@ -45,9 +45,13 @@ class TS:
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['ds'] = pd.to_datetime(pdf['ds'])
50
- pdf['y'] = data.close.values
 
 
 
 
51
  return pdf
52
  def prophet(self, period = 28):
53
  df = self.get_data()
@@ -58,10 +62,14 @@ class TS:
58
  return self.viz(df, forecast)
59
  def viz(self, data, future):
60
  fig = go.Figure()
61
- fig.add_trace(go.Scatter(x= data.ds,
62
- y=data.y,
63
- name = f"{self.symbol}"
64
- ))
 
 
 
 
65
  fig.add_trace(go.Scatter(x= future.ds,
66
  y=future.yhat,
67
  name = f"{self.symbol}_pred"
 
45
  start_date=start_date,
46
  end_date=end_date, resolution='1D', type='stock')
47
  pdf = pd.DataFrame()
48
+ pdf['ds'] = data["time"]
49
  pdf['ds'] = pd.to_datetime(pdf['ds'])
50
+ pdf["open"] = data["open"]
51
+ pdf["close"] = data["close"]
52
+ pdf["high"] = data["high"]
53
+ pdf["low"] = data["low"]
54
+ pdf['y'] = data.close
55
  return pdf
56
  def prophet(self, period = 28):
57
  df = self.get_data()
 
62
  return self.viz(df, forecast)
63
  def viz(self, data, future):
64
  fig = go.Figure()
65
+ fig.add_trace(go.Candlestick(x = data["time"],
66
+ open = data["open"],
67
+ high = data["high"],
68
+ low = data["low"],
69
+ close = data["close"]
70
+ name = f"{self.symbol}"
71
+
72
+ ))
73
  fig.add_trace(go.Scatter(x= future.ds,
74
  y=future.yhat,
75
  name = f"{self.symbol}_pred"