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()
|