changed labels

get_forecast.py

@@ -48,18 +48,18 @@ def get_forecast(serie: str, periods, percent_change: int):
     aux_0 = pd.concat([last_obs, values_0])
     
     # arrange dataframe
-    df_future = future_values.rename(columns = {'ds': 'Date', 'yhat': f'{serie} with price change'}).copy()
+    df_future = future_values.rename(columns = {'ds': 'Date', 'yhat': f'{serie} with price policy change'}).copy()
     df_future['Date'] = df_future['Date'].apply(lambda x: x.date())
-    df_future[f'{serie} without price change'] = values_0.rename(columns = {'ds': 'Date', 'yhat': serie})[serie]
-    df_future['Diff'] = df_future[f'{serie} with price change'] - df_future[f'{serie} without price change']
-    df_future[f'{serie} with price change'] = df_future[f'{serie} with price change'].apply(lambda x: round(x, 4))
-    df_future[f'{serie} without price change'] = df_future[f'{serie} without price change'].apply(lambda x: round(x, 4))
+    df_future[f'{serie} without policy change'] = values_0.rename(columns = {'ds': 'Date', 'yhat': serie})[serie]
+    df_future['Diff'] = df_future[f'{serie} with price policy change'] - df_future[f'{serie} without policy change']
+    df_future[f'{serie} with price policy change'] = df_future[f'{serie} with price policy change'].apply(lambda x: round(x, 4))
+    df_future[f'{serie} without policy change'] = df_future[f'{serie} without policy change'].apply(lambda x: round(x, 4))
     df_future['Diff'] = df_future['Diff'].apply(lambda x: round(x, 4))
 
     # plot
     fig = plt.figure()
-    plt.plot(future_aux['ds'], future_aux['yhat'], label = 'Price Change Forecast', marker = '.', color = 'C1')
-    plt.plot(aux_0['ds'], aux_0['yhat'], label = 'No Price Change Forecast', marker = '.', color = 'C2')
+    plt.plot(future_aux['ds'], future_aux['yhat'], label = 'Price Policy Change Forecast', marker = '.', color = 'C1')
+    plt.plot(aux_0['ds'], aux_0['yhat'], label = 'No Policy Change Forecast', marker = '.', color = 'C2')
     plt.plot(history['ds'], history['y'], label = 'Historic data', marker = '.', color = 'C0')
     plt.xticks(rotation = 45)
     plt.ylabel(serie)