leuschnm commited on
Commit
a4f898a
·
1 Parent(s): 06db282
Files changed (1) hide show
  1. app.py +7 -4
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].scatter(df.loc[df['Group'] == '4', 'Date'], df.loc[df['Group'] == '4', 'sales'], color='grey', marker='o')
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].scatter(df.loc[df['Group'] == '7', 'Date'], df.loc[df['Group'] == '7', 'sales'], color='blue', marker='o')
 
74
  axs[0, 1].set_title('Article Group 2')
75
 
76
- axs[1, 0].scatter(df.loc[df['Group'] == '1', 'Date'], df.loc[df['Group'] == '1', 'sales'], color='green', marker='o')
 
77
  axs[1, 0].set_title('Article Group 3')
78
 
79
- axs[1, 1].scatter(df.loc[df['Group'] == '6', 'Date'], df.loc[df['Group'] == '6', 'sales'], color='yellow', marker='o')
 
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