azrai99 commited on
Commit
a709b08
·
verified ·
1 Parent(s): 00cddc5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +24 -18
app.py CHANGED
@@ -207,11 +207,13 @@ def transfer_learning_forecasting():
207
  st.session_state.df = df
208
 
209
  columns = df.columns.tolist() # Convert Index to list
210
- if 'ds' in columns and 'unique_id' in columns:
211
- options = columns.remove('ds')
212
- options.remove('unique_id')
213
  ds_col = st.selectbox("Select Date/Time column", options=columns, index=columns.index('ds') if 'ds' in columns else 0)
214
- y_col = st.selectbox("Select Target column", options=options, index=0)
 
 
 
 
215
 
216
  st.session_state.ds_col = ds_col
217
  st.session_state.y_col = y_col
@@ -290,11 +292,13 @@ def dynamic_forecasting():
290
  st.session_state.df = df
291
 
292
  columns = df.columns.tolist() # Convert Index to list
293
- if 'ds' in columns and 'unique_id' in columns:
294
- options = columns.remove('ds')
295
- options.remove('unique_id')
296
  ds_col = st.selectbox("Select Date/Time column", options=columns, index=columns.index('ds') if 'ds' in columns else 0)
297
- y_col = st.selectbox("Select Target column", options=options, index=0)
 
 
 
 
298
 
299
  df = df.rename(columns={ds_col: 'ds', y_col: 'y'})
300
 
@@ -331,13 +335,14 @@ def timegpt_fcst():
331
  df = load_default()
332
  st.session_state.df = df
333
 
334
- # Column selection
335
  columns = df.columns.tolist() # Convert Index to list
336
- if 'ds' in columns and 'unique_id' in columns:
337
- options = columns.remove('ds')
338
- options.remove('unique_id')
339
  ds_col = st.selectbox("Select Date/Time column", options=columns, index=columns.index('ds') if 'ds' in columns else 0)
340
- y_col = st.selectbox("Select Target column", options=options, index=0)
 
 
 
 
341
 
342
  df = df.rename(columns={ds_col: 'ds', y_col: 'y'})
343
 
@@ -375,13 +380,14 @@ def timegpt_anom():
375
  df = load_default()
376
  st.session_state.df = df
377
 
378
- # Column selection
379
  columns = df.columns.tolist() # Convert Index to list
380
- if 'ds' in columns and 'unique_id' in columns:
381
- options = columns.remove('ds')
382
- options.remove('unique_id')
383
  ds_col = st.selectbox("Select Date/Time column", options=columns, index=columns.index('ds') if 'ds' in columns else 0)
384
- y_col = st.selectbox("Select Target column", options=options, index=0)
 
 
 
 
385
 
386
  df = df.rename(columns={ds_col: 'ds', y_col: 'y'})
387
 
 
207
  st.session_state.df = df
208
 
209
  columns = df.columns.tolist() # Convert Index to list
210
+ opt = []
 
 
211
  ds_col = st.selectbox("Select Date/Time column", options=columns, index=columns.index('ds') if 'ds' in columns else 0)
212
+ if 'ds' in columns and 'unique_id' in columns:
213
+ columns.pop('ds')
214
+ columns.pop('unique_id')
215
+ opt = columns
216
+ y_col = st.selectbox("Select Target column", options=opt, index=0)
217
 
218
  st.session_state.ds_col = ds_col
219
  st.session_state.y_col = y_col
 
292
  st.session_state.df = df
293
 
294
  columns = df.columns.tolist() # Convert Index to list
295
+ opt = []
 
 
296
  ds_col = st.selectbox("Select Date/Time column", options=columns, index=columns.index('ds') if 'ds' in columns else 0)
297
+ if 'ds' in columns and 'unique_id' in columns:
298
+ columns.pop('ds')
299
+ columns.pop('unique_id')
300
+ opt = columns
301
+ y_col = st.selectbox("Select Target column", options=opt, index=0)
302
 
303
  df = df.rename(columns={ds_col: 'ds', y_col: 'y'})
304
 
 
335
  df = load_default()
336
  st.session_state.df = df
337
 
 
338
  columns = df.columns.tolist() # Convert Index to list
339
+ opt = []
 
 
340
  ds_col = st.selectbox("Select Date/Time column", options=columns, index=columns.index('ds') if 'ds' in columns else 0)
341
+ if 'ds' in columns and 'unique_id' in columns:
342
+ columns.pop('ds')
343
+ columns.pop('unique_id')
344
+ opt = columns
345
+ y_col = st.selectbox("Select Target column", options=opt, index=0)
346
 
347
  df = df.rename(columns={ds_col: 'ds', y_col: 'y'})
348
 
 
380
  df = load_default()
381
  st.session_state.df = df
382
 
 
383
  columns = df.columns.tolist() # Convert Index to list
384
+ opt = []
 
 
385
  ds_col = st.selectbox("Select Date/Time column", options=columns, index=columns.index('ds') if 'ds' in columns else 0)
386
+ if 'ds' in columns and 'unique_id' in columns:
387
+ columns.pop('ds')
388
+ columns.pop('unique_id')
389
+ opt = columns
390
+ y_col = st.selectbox("Select Target column", options=opt, index=0)
391
 
392
  df = df.rename(columns={ds_col: 'ds', y_col: 'y'})
393