Spaces:
Runtime error
Runtime error
bug fix
Browse files
app.py
CHANGED
|
@@ -66,17 +66,20 @@ def generate_plot(df, preds):
|
|
| 66 |
df = pd.merge(df, preds, left_on=["time_idx", "Group"], right_on=["pred_idx", "Group"], how = "left")
|
| 67 |
df[["sales", "pred"]] = df[["sales", "pred"]].replace(0.0, np.nan)
|
| 68 |
# Plot scatter plots for each group
|
| 69 |
-
axs[0, 0].
|
| 70 |
axs[0, 0].plot(df.loc[df['Group'] == '4', 'Date'], df.loc[df['Group'] == '4', 'pred'], color = 'red')
|
| 71 |
axs[0, 0].set_title('Article Group 1')
|
| 72 |
|
| 73 |
-
axs[0, 1].
|
|
|
|
| 74 |
axs[0, 1].set_title('Article Group 2')
|
| 75 |
|
| 76 |
-
axs[1, 0].
|
|
|
|
| 77 |
axs[1, 0].set_title('Article Group 3')
|
| 78 |
|
| 79 |
-
axs[1, 1].
|
|
|
|
| 80 |
axs[1, 1].set_title('Article Group 4')
|
| 81 |
|
| 82 |
# Adjust spacing between subplots
|
|
|
|
| 66 |
df = pd.merge(df, preds, left_on=["time_idx", "Group"], right_on=["pred_idx", "Group"], how = "left")
|
| 67 |
df[["sales", "pred"]] = df[["sales", "pred"]].replace(0.0, np.nan)
|
| 68 |
# Plot scatter plots for each group
|
| 69 |
+
axs[0, 0].plot(df.loc[df['Group'] == '4', 'Date'], df.loc[df['Group'] == '4', 'sales'], color='grey')
|
| 70 |
axs[0, 0].plot(df.loc[df['Group'] == '4', 'Date'], df.loc[df['Group'] == '4', 'pred'], color = 'red')
|
| 71 |
axs[0, 0].set_title('Article Group 1')
|
| 72 |
|
| 73 |
+
axs[0, 1].plot(df.loc[df['Group'] == '7', 'Date'], df.loc[df['Group'] == '7', 'sales'], color='grey')
|
| 74 |
+
axs[0, 1].plot(df.loc[df['Group'] == '7', 'Date'], df.loc[df['Group'] == '4', 'pred'], color = 'red')
|
| 75 |
axs[0, 1].set_title('Article Group 2')
|
| 76 |
|
| 77 |
+
axs[1, 0].plot(df.loc[df['Group'] == '1', 'Date'], df.loc[df['Group'] == '1', 'sales'], color='grey')
|
| 78 |
+
axs[1, 0].plot(df.loc[df['Group'] == '1', 'Date'], df.loc[df['Group'] == '4', 'pred'], color = 'red')
|
| 79 |
axs[1, 0].set_title('Article Group 3')
|
| 80 |
|
| 81 |
+
axs[1, 1].plot(df.loc[df['Group'] == '6', 'Date'], df.loc[df['Group'] == '6', 'sales'], color='grey')
|
| 82 |
+
axs[1, 1].plot(df.loc[df['Group'] == '6', 'Date'], df.loc[df['Group'] == '4', 'pred'], color = 'red')
|
| 83 |
axs[1, 1].set_title('Article Group 4')
|
| 84 |
|
| 85 |
# Adjust spacing between subplots
|