tbdavid2019 commited on
Commit
843b7e5
·
1 Parent(s): 6d62996
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -26,10 +26,12 @@ def prepare_data_chronos(data):
26
  formatted_df = formatted_df.sort_values('timestamp')
27
 
28
  try:
29
- # 指定頻率創建 TimeSeriesDataFrame
30
- ts_df = TimeSeriesDataFrame(
31
  formatted_df,
32
- freq="1D" # 指定為每日數據
 
 
33
  )
34
  return ts_df
35
  except Exception as e:
@@ -115,10 +117,9 @@ def get_top_10_potential_stocks(period, selected_indices):
115
 
116
  ts_data = prepare_data_chronos(data)
117
 
118
- # 創建預測器時指定頻率
119
  predictor = TimeSeriesPredictor(
120
  prediction_length=prediction_length,
121
- target='value',
122
  freq="1D" # 指定為每日數據
123
  )
124
 
 
26
  formatted_df = formatted_df.sort_values('timestamp')
27
 
28
  try:
29
+ # 使用 from_data_frame 方法創建 TimeSeriesDataFrame
30
+ ts_df = TimeSeriesDataFrame.from_data_frame(
31
  formatted_df,
32
+ id_column='item_id',
33
+ timestamp_column='timestamp',
34
+ target_column='value'
35
  )
36
  return ts_df
37
  except Exception as e:
 
117
 
118
  ts_data = prepare_data_chronos(data)
119
 
120
+ # 只在創建預測器時指定頻率
121
  predictor = TimeSeriesPredictor(
122
  prediction_length=prediction_length,
 
123
  freq="1D" # 指定為每日數據
124
  )
125