azrai99 commited on
Commit
c875dd7
·
verified ·
1 Parent(s): b0365ad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -2
app.py CHANGED
@@ -228,8 +228,8 @@ def transfer_learning_forecasting():
228
 
229
  col1, col2 = st.columns([2,4])
230
  with col1:
231
- tab_insample = st.tabs(
232
- ["Input data"]
233
  )
234
  with tab_insample:
235
  df_grid = df.drop(columns="unique_id")
@@ -240,6 +240,15 @@ def transfer_learning_forecasting():
240
  height=360,
241
  )
242
 
 
 
 
 
 
 
 
 
 
243
  with col2:
244
  # Load pre-trained models
245
  nhits_model, timesnet_model, lstm_model, tft_model = select_model_based_on_frequency(frequency, nhits_models, timesnet_models, lstm_models, tft_models)
 
228
 
229
  col1, col2 = st.columns([2,4])
230
  with col1:
231
+ tab_insample, tab_forecast = st.tabs(
232
+ ["Input data", "Forecast"]
233
  )
234
  with tab_insample:
235
  df_grid = df.drop(columns="unique_id")
 
240
  height=360,
241
  )
242
 
243
+ with tab_forecast:
244
+ df_grid = df.drop(columns="unique_id")
245
+ grid_table = AgGrid(
246
+ df_grid,
247
+ theme="alpine",
248
+ fit_columns_on_grid_load=True,
249
+ height=360,
250
+ )
251
+
252
  with col2:
253
  # Load pre-trained models
254
  nhits_model, timesnet_model, lstm_model, tft_model = select_model_based_on_frequency(frequency, nhits_models, timesnet_models, lstm_models, tft_models)