azrai99 commited on
Commit
81f859b
·
verified ·
1 Parent(s): 752c404

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -266,7 +266,7 @@ def transfer_learning_forecasting():
266
  forecast_results['TFT'] = generate_forecast(tft_model, df)
267
 
268
  for model_name, forecast_df in forecast_results.items():
269
- plot_forecasts(forecast_df, df, f'{model_name} Forecast for {y_col}')
270
 
271
  end_time = time.time() # End timing
272
  time_taken = end_time - start_time
 
266
  forecast_results['TFT'] = generate_forecast(tft_model, df)
267
 
268
  for model_name, forecast_df in forecast_results.items():
269
+ plot_forecasts(forecast_df.iloc[:horizon,:], df, f'{model_name} Forecast for {y_col}')
270
 
271
  end_time = time.time() # End timing
272
  time_taken = end_time - start_time