kashif HF staff commited on
Commit
e460697
1 Parent(s): baa97f0

sort date index

Browse files
Files changed (1) hide show
  1. app.py +21 -3
app.py CHANGED
@@ -14,23 +14,41 @@ def offset_calculation(prediction_length, rolling_windows, length):
14
  return row_offset
15
 
16
 
17
- def preprocess(input_data, prediction_length, rolling_windows, progress=gr.Progress(track_tqdm=True)):
 
 
 
 
 
 
18
  df = pd.read_csv(input_data.name, index_col=0, parse_dates=True)
 
19
  row_offset = offset_calculation(prediction_length, rolling_windows, len(df))
20
  return plot_train_test(df.iloc[:row_offset], df.iloc[row_offset:])
21
 
22
 
23
- def train_and_forecast(input_data, prediction_length, rolling_windows, epochs, progress=gr.Progress(track_tqdm=True)):
 
 
 
 
 
 
24
  if not input_data:
25
  raise gr.Error("Upload a file with the Upload button")
26
  try:
27
  df = pd.read_csv(input_data.name, index_col=0, parse_dates=True)
 
28
  except AttributeError:
29
  raise gr.Error("Upload a file with the Upload button")
30
 
31
  row_offset = offset_calculation(prediction_length, rolling_windows, len(df))
32
 
33
- gluon_df = PandasDataset(df, target=df.columns[0])
 
 
 
 
34
 
35
  training_data, test_gen = split(gluon_df, offset=row_offset)
36
 
 
14
  return row_offset
15
 
16
 
17
+ def preprocess(
18
+ input_data,
19
+ prediction_length,
20
+ rolling_windows,
21
+ item_id,
22
+ progress=gr.Progress(track_tqdm=True),
23
+ ):
24
  df = pd.read_csv(input_data.name, index_col=0, parse_dates=True)
25
+ df.sort_index(inplace=True)
26
  row_offset = offset_calculation(prediction_length, rolling_windows, len(df))
27
  return plot_train_test(df.iloc[:row_offset], df.iloc[row_offset:])
28
 
29
 
30
+ def train_and_forecast(
31
+ input_data,
32
+ prediction_length,
33
+ rolling_windows,
34
+ epochs,
35
+ progress=gr.Progress(track_tqdm=True),
36
+ ):
37
  if not input_data:
38
  raise gr.Error("Upload a file with the Upload button")
39
  try:
40
  df = pd.read_csv(input_data.name, index_col=0, parse_dates=True)
41
+ df.sort_index(inplace=True)
42
  except AttributeError:
43
  raise gr.Error("Upload a file with the Upload button")
44
 
45
  row_offset = offset_calculation(prediction_length, rolling_windows, len(df))
46
 
47
+ try:
48
+ gluon_df = PandasDataset(df, target=df.columns[0])
49
+ except TypeError:
50
+ freq = pd.infer_freq(df.index[:3])
51
+ gluon_df = PandasDataset(df, target=df.columns[0], freq=freq)
52
 
53
  training_data, test_gen = split(gluon_df, offset=row_offset)
54