tbdavid2019 commited on
Commit
9c0b92a
·
1 Parent(s): 1d22072
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -19,14 +19,14 @@ def prepare_data_chronos(data):
19
  formatted_df = pd.DataFrame({
20
  'item_id': ['stock'] * len(df),
21
  'timestamp': pd.to_datetime(df['Date']),
22
- 'value': df['Close'].astype('float32').values.ravel()
23
  })
24
 
25
  # 按照時間戳排序
26
  formatted_df = formatted_df.sort_values('timestamp')
27
 
28
  try:
29
- # 只使用必要的參數
30
  ts_df = TimeSeriesDataFrame.from_data_frame(
31
  formatted_df,
32
  id_column='item_id',
@@ -116,10 +116,11 @@ def get_top_10_potential_stocks(period, selected_indices):
116
 
117
  ts_data = prepare_data_chronos(data)
118
 
119
- # 只在創建預測器時指定頻率
120
  predictor = TimeSeriesPredictor(
121
  prediction_length=prediction_length,
122
- freq="1D" # 指定為每日數據
 
123
  )
124
 
125
  predictor.fit(
 
19
  formatted_df = pd.DataFrame({
20
  'item_id': ['stock'] * len(df),
21
  'timestamp': pd.to_datetime(df['Date']),
22
+ 'target': df['Close'].astype('float32').values.ravel() # 改回使用 'target' 而不是 'value'
23
  })
24
 
25
  # 按照時間戳排序
26
  formatted_df = formatted_df.sort_values('timestamp')
27
 
28
  try:
29
+ # 創建 TimeSeriesDataFrame
30
  ts_df = TimeSeriesDataFrame.from_data_frame(
31
  formatted_df,
32
  id_column='item_id',
 
116
 
117
  ts_data = prepare_data_chronos(data)
118
 
119
+ # 創建預測器,使用 'target' 作為目標列名
120
  predictor = TimeSeriesPredictor(
121
  prediction_length=prediction_length,
122
+ freq="D", # 使用 'D' 而不是 '1D'
123
+ target="target" # 確保與數據框中的列名一致
124
  )
125
 
126
  predictor.fit(