stinoco commited on
Commit
630939d
·
1 Parent(s): 2f96ffc

added 0 price change scenario

Browse files
Files changed (1) hide show
  1. get_forecast.py +9 -5
get_forecast.py CHANGED
@@ -35,11 +35,6 @@ def get_forecast(serie: str, periods, percent_change: int):
35
  forecast = model.predict(future)
36
 
37
  future_values = forecast[['ds', 'yhat']]
38
-
39
- # arrange dataframe
40
- df_future = future_values.rename(columns = {'ds': 'Date', 'yhat': serie}).copy()
41
- df_future['Date'] = df_future['Date'].apply(lambda x: x.date())
42
- df_future[serie] = df_future[serie].apply(lambda x: round(x, 4))
43
 
44
  # aux to plot
45
  last_obs = history.iloc[-1:][['ds', 'y']].rename(columns = {'y': 'yhat'})
@@ -51,6 +46,15 @@ def get_forecast(serie: str, periods, percent_change: int):
51
  forecast_0 = model.predict(future_0)
52
  values_0 = forecast_0[['ds', 'yhat']]
53
  aux_0 = pd.concat([last_obs, values_0])
 
 
 
 
 
 
 
 
 
54
 
55
  # plot
56
  fig = plt.figure()
 
35
  forecast = model.predict(future)
36
 
37
  future_values = forecast[['ds', 'yhat']]
 
 
 
 
 
38
 
39
  # aux to plot
40
  last_obs = history.iloc[-1:][['ds', 'y']].rename(columns = {'y': 'yhat'})
 
46
  forecast_0 = model.predict(future_0)
47
  values_0 = forecast_0[['ds', 'yhat']]
48
  aux_0 = pd.concat([last_obs, values_0])
49
+
50
+ # arrange dataframe
51
+ df_future = future_values.rename(columns = {'ds': 'Date', 'yhat': serie}).copy()
52
+ df_future['Date'] = df_future['Date'].apply(lambda x: x.date())
53
+ df_future[f'No change {serie}'] = values_0.rename(columns = {'ds': 'Date', 'yhat': serie})[serie]
54
+ df_future['Diff'] = df_future[serie] - df_future[f'No change {serie}']
55
+ df_future[serie] = df_future[serie].apply(lambda x: round(x, 4))
56
+ df_future[f'No change {serie}'] = df_future[f'No change {serie}'].apply(lambda x: round(x, 4))
57
+ df_future['Diff'] = df_future['Diff'].apply(lambda x: round(x, 4))
58
 
59
  # plot
60
  fig = plt.figure()