From 063f1adac626f778ceb287e7c7aae8d433854e1a Mon Sep 17 00:00:00 2001 From: "Dr. Kashif Rasul" Date: Wed, 2 Sep 2020 15:00:37 +0200 Subject: [PATCH] fix ordering of feat_static_cat --- pts/dataset/repository/_m5.py | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/pts/dataset/repository/_m5.py b/pts/dataset/repository/_m5.py index 8ec52bd..aa25826 100644 --- a/pts/dataset/repository/_m5.py +++ b/pts/dataset/repository/_m5.py @@ -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)