Spaces:
Runtime error
Runtime error
added 0 price change scenario
Browse files- 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()
|