changed labels

get_forecast.py

@@ -53,13 +53,13 @@ def get_forecast(serie: str, periods, percent_change: int):
     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'{serie} without prince change'] = df_future[f'{serie} without 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['Diff'] = df_future['Diff'].apply(lambda x: round(x, 4))
 
     # plot
     fig = plt.figure()
-    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(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(history['ds'], history['y'], label = 'Historic data', marker = '.', color = 'C0')
     plt.xticks(rotation = 45)
     plt.ylabel(serie)