tbdavid2019 commited on
Commit
4fe64cc
·
1 Parent(s): 262820f
Files changed (2) hide show
  1. .gitignore +1 -0
  2. app.py +7 -15
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ myenv/
app.py CHANGED
@@ -16,29 +16,21 @@ def prepare_data_chronos(data):
16
  data = data.reset_index()
17
  data = data.rename(columns={"Date": "timestamp", "Close": "target"})
18
 
19
- # 只保留需要的欄位
20
  data = data[["timestamp", "target"]]
 
21
 
22
  # 設定正確的資料類型
23
- data = data.astype({
24
- "timestamp": "datetime64[ns]",
25
- "target": "float32"
26
- })
27
-
28
- # 添加 item_id
29
- data["item_id"] = "stock"
30
 
31
- # 建立 TimeSeriesDataFrame 並指定資料類型
32
- ts_data = TimeSeriesDataFrame(
33
  data,
34
  id_column="item_id",
35
- timestamp_column="timestamp",
36
- target_column="target"
37
  )
38
 
39
- # 確保時間序列資料是按時間排序的
40
- ts_data = ts_data.sort_index()
41
-
42
  return ts_data
43
 
44
 
 
16
  data = data.reset_index()
17
  data = data.rename(columns={"Date": "timestamp", "Close": "target"})
18
 
19
+ # 只保留需要的欄位並設定正確的資料類型
20
  data = data[["timestamp", "target"]]
21
+ data["item_id"] = "stock"
22
 
23
  # 設定正確的資料類型
24
+ data["timestamp"] = pd.to_datetime(data["timestamp"])
25
+ data["target"] = data["target"].astype('float32')
 
 
 
 
 
26
 
27
+ # 建立 TimeSeriesDataFrame,只使用必要的參數
28
+ ts_data = TimeSeriesDataFrame.from_data_frame(
29
  data,
30
  id_column="item_id",
31
+ timestamp_column="timestamp"
 
32
  )
33
 
 
 
 
34
  return ts_data
35
 
36