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
|