kaxap commited on
Commit
6c7702c
·
1 Parent(s): 6c7af43

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -12
app.py CHANGED
@@ -56,18 +56,18 @@ def train_and_forecast(
56
  except AttributeError:
57
  raise gr.Error("Upload a file with the Upload button")
58
 
59
- # freq = pd.infer_freq(df.index[:3])
60
- # print(freq)
61
- # date_range = pd.date_range(df.index[0], pd.DateOffset(months=prediction_length//2) + df.index[-1], freq=freq)
62
- # print(date_range)
63
- # new_df = df.reindex(date_range)
64
- # new_df = new_df.interpolate(method='spline', order=2)
65
- # # trying this hack to fix pandas beginMonth stuff
66
- # buffer = io.BytesIO()
67
- # new_df.to_csv(buffer, index=False)
68
- # buffer.seek(0)
69
- # df = pd.read_csv(buffer)
70
- # df.sort_index(inplace=True)
71
  print(df)
72
  gluon_df = PandasDataset(df, target=df.columns[0])
73
  print(gluon_df)
 
56
  except AttributeError:
57
  raise gr.Error("Upload a file with the Upload button")
58
 
59
+ freq = pd.infer_freq(df.index[:3])
60
+ print(freq)
61
+ date_range = pd.date_range(df.index[0], pd.DateOffset(months=prediction_length//2) + df.index[-1], freq=freq)
62
+ print(date_range)
63
+ new_df = df.reindex(date_range)
64
+ new_df = new_df.interpolate(method='spline', order=2)
65
+ # trying this hack to fix pandas beginMonth stuff
66
+ buffer = io.BytesIO()
67
+ new_df.to_csv(buffer, index=False)
68
+ buffer.seek(0)
69
+ df = pd.read_csv(buffer, index_col=0, parse_dates=True)
70
+ df.sort_index(inplace=True)
71
  print(df)
72
  gluon_df = PandasDataset(df, target=df.columns[0])
73
  print(gluon_df)