Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|