Update app.py
Browse files
app.py
CHANGED
|
@@ -171,9 +171,9 @@ def plot(y, label, timeframe='1h', ma=5, n_steps=None):
|
|
| 171 |
plt.figure(figsize=(12,6))
|
| 172 |
if ma:
|
| 173 |
df_plot = pd.DataFrame({'date': label['timestamp'].values[-n_steps:], 'prediction':5*(y[-n_steps:]-0.5), 'real': label['close'].values[-n_steps:]})
|
| 174 |
-
plt.plot(df_plot['date'], df_plot['prediction'].rolling(window=ma).mean(), marker='o',
|
| 175 |
-
plt.plot(df_plot['date'], df_plot['real'].rolling(window=ma).mean(), marker='o',
|
| 176 |
-
plt.plot(df_plot['date'], (df_plot['real']-df_plot['prediction']).rolling(window=ma).mean(), marker='o',
|
| 177 |
else:
|
| 178 |
plt.plot(label['timestamp'].values[-n_steps:], 5*(y[-n_steps:]-0.5), label='updown')
|
| 179 |
plt.plot(label['timestamp'].values[-n_steps:], label['close'].values[-n_steps:], label='real')
|
|
|
|
| 171 |
plt.figure(figsize=(12,6))
|
| 172 |
if ma:
|
| 173 |
df_plot = pd.DataFrame({'date': label['timestamp'].values[-n_steps:], 'prediction':5*(y[-n_steps:]-0.5), 'real': label['close'].values[-n_steps:]})
|
| 174 |
+
plt.plot(df_plot['date'], df_plot['prediction'].rolling(window=ma).mean(), marker='o', label='updown')
|
| 175 |
+
plt.plot(df_plot['date'], df_plot['real'].rolling(window=ma).mean(), marker='o', label='real')
|
| 176 |
+
plt.plot(df_plot['date'], (df_plot['real']-df_plot['prediction']).rolling(window=ma).mean(), marker='o', label='difference')
|
| 177 |
else:
|
| 178 |
plt.plot(label['timestamp'].values[-n_steps:], 5*(y[-n_steps:]-0.5), label='updown')
|
| 179 |
plt.plot(label['timestamp'].values[-n_steps:], label['close'].values[-n_steps:], label='real')
|