wassname/pytorch-ts
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-ts.git synced 2026-08-13 12:30:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix ordering of feat_static_cat

Browse Source
This commit is contained in:
Dr. Kashif Rasul
2020-09-02 15:00:37 +02:00
parent 262d40026a
commit 063f1adac6
1 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/dataset/repository/_m5.py
+10 -10
View File
@@ -141,17 +141,17 @@ def generate_m5_dataset(
start_date = time_index[start_index]
time_series = {}
store_enc, state_enc, cat_enc, dept_enc, item_enc = item.iloc[1913:]
state_enc, store_enc, cat_enc, dept_enc, item_enc = item.iloc[1913:]
time_series["start"] = str(start_date)
time_series["item_id"] = id[:-11]
time_series["feat_static_cat"] = [
item_enc,
dept_enc,
cat_enc,
store_enc,
state_enc,
store_enc,
cat_enc,
dept_enc,
item_enc,
]
sell_price = get_sell_price(item_id, store_id)
@@ -202,17 +202,17 @@ def generate_m5_dataset(
start_date = time_index[start_index]
time_series = {}
store_enc, state_enc, cat_enc, dept_enc, item_enc = item.iloc[1941:]
state_enc, store_enc, cat_enc, dept_enc, item_enc = item.iloc[1941:]
time_series["start"] = str(start_date)
time_series["item_id"] = id[:-11]
time_series["feat_static_cat"] = [
item_enc,
dept_enc,
cat_enc,
store_enc,
state_enc,
store_enc,
cat_enc,
dept_enc,
item_enc,
]
sell_price = get_sell_price(item_id, store_id)