changed labels

get_forecast.py

@@ -50,19 +50,18 @@ def get_forecast(serie: str, periods, percent_change: int):
     # arrange dataframe
     df_future = future_values.rename(columns = {'ds': 'Date', 'yhat': serie}).copy()
     df_future['Date'] = df_future['Date'].apply(lambda x: x.date())
-    df_future[f'No change {serie}'] = values_0.rename(columns = {'ds': 'Date', 'yhat': serie})[serie]
-    df_future['Diff'] = df_future[serie] - df_future[f'No change {serie}']
+    df_future[f'{serie} without price change'] = values_0.rename(columns = {'ds': 'Date', 'yhat': serie})[serie]
+    df_future['Diff'] = df_future[serie] - df_future[f'{serie} without price change']
     df_future[serie] = df_future[serie].apply(lambda x: round(x, 4))
-    df_future[f'No change {serie}'] = df_future[f'No change {serie}'].apply(lambda x: round(x, 4))
+    df_future[f'{serie} without prince change'] = df_future[f'{serie} without price 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 = 'Forecast', marker = '.', color = 'C1')
-    plt.plot(aux_0['ds'], aux_0['yhat'], label = 'No change', marker = '.', color = 'C2')
+    plt.plot(future_aux['ds'], future_aux['yhat'], label = 'Forecast with price change', marker = '.', color = 'C1')
+    plt.plot(aux_0['ds'], aux_0['yhat'], label = 'Forecast without price change', marker = '.', color = 'C2')
     plt.plot(history['ds'], history['y'], label = 'Historic data', marker = '.', color = 'C0')
     plt.xticks(rotation = 45)
-    #plt.xlabel('Date')
     plt.ylabel(serie)
     plt.tight_layout()
     plt.legend()