change vnquant to vnstock
Browse files
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
|
49 |
pdf['ds'] = pd.to_datetime(pdf['ds'])
|
50 |
-
pdf[
|
|
|
|
|
|
|
|
|
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.
|
62 |
-
|
63 |
-
|
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"
|