Fix embeddings sizes.

8 files changed, 16 insertions, 16 deletions

diff --git a/config/joint_simple_mlp_tgtcls_111_cswdtx.py b/config/joint_simple_mlp_tgtcls_111_cswdtx.py
index 6c58312..a18e4ff 100644
--- a/config/joint_simple_mlp_tgtcls_111_cswdtx.py
+++ b/config/joint_simple_mlp_tgtcls_111_cswdtx.py
@@ -20,8 +20,8 @@ for i in range(22):
     time_tgtcls.append(time_tgtcls[-1] + time_tgtcls[-2])
 
 dim_embeddings = [
-    ('origin_call', data.origin_call_size+1, 10),
-    ('origin_stand', data.stands_size+1, 10),
+    ('origin_call', data.origin_call_size, 10),
+    ('origin_stand', data.stands_size, 10),
     ('week_of_year', 52, 10),
     ('day_of_week', 7, 10),
     ('qhour_of_day', 24 * 4, 10),
diff --git a/config/joint_simple_mlp_tgtcls_111_cswdtx_bigger.py b/config/joint_simple_mlp_tgtcls_111_cswdtx_bigger.py
index 527b803..0403197 100644
--- a/config/joint_simple_mlp_tgtcls_111_cswdtx_bigger.py
+++ b/config/joint_simple_mlp_tgtcls_111_cswdtx_bigger.py
@@ -20,8 +20,8 @@ for i in range(21):
     time_tgtcls.append(time_tgtcls[-1] + time_tgtcls[-2])
 
 dim_embeddings = [
-    ('origin_call', data.origin_call_size+1, 15),
-    ('origin_stand', data.stands_size+1, 10),
+    ('origin_call', data.origin_call_size, 15),
+    ('origin_stand', data.stands_size, 10),
     ('week_of_year', 52, 10),
     ('day_of_week', 7, 10),
     ('qhour_of_day', 24 * 4, 10),
diff --git a/config/joint_simple_mlp_tgtcls_111_cswdtx_bigger_dropout.py b/config/joint_simple_mlp_tgtcls_111_cswdtx_bigger_dropout.py
index a6af132..937122d 100644
--- a/config/joint_simple_mlp_tgtcls_111_cswdtx_bigger_dropout.py
+++ b/config/joint_simple_mlp_tgtcls_111_cswdtx_bigger_dropout.py
@@ -20,8 +20,8 @@ for i in range(21):
     time_tgtcls.append(time_tgtcls[-1] + time_tgtcls[-2])
 
 dim_embeddings = [
-    ('origin_call', data.origin_call_size+1, 15),
-    ('origin_stand', data.stands_size+1, 10),
+    ('origin_call', data.origin_call_size, 15),
+    ('origin_stand', data.stands_size, 10),
     ('week_of_year', 52, 10),
     ('day_of_week', 7, 10),
     ('qhour_of_day', 24 * 4, 10),
diff --git a/config/joint_simple_mlp_tgtcls_111_cswdtx_noise_dout.py b/config/joint_simple_mlp_tgtcls_111_cswdtx_noise_dout.py
index c3686e6..6d44c10 100644
--- a/config/joint_simple_mlp_tgtcls_111_cswdtx_noise_dout.py
+++ b/config/joint_simple_mlp_tgtcls_111_cswdtx_noise_dout.py
@@ -23,8 +23,8 @@ for i in range(22):
     time_tgtcls.append(time_tgtcls[-1] + time_tgtcls[-2])
 
 dim_embeddings = [
-    ('origin_call', data.origin_call_size+1, 10),
-    ('origin_stand', data.stands_size+1, 10),
+    ('origin_call', data.origin_call_size, 10),
+    ('origin_stand', data.stands_size, 10),
     ('week_of_year', 52, 10),
     ('day_of_week', 7, 10),
     ('qhour_of_day', 24 * 4, 10),
diff --git a/config/joint_simple_mlp_tgtcls_1_cswdtx.py b/config/joint_simple_mlp_tgtcls_1_cswdtx.py
index c1bab07..1874444 100644
--- a/config/joint_simple_mlp_tgtcls_1_cswdtx.py
+++ b/config/joint_simple_mlp_tgtcls_1_cswdtx.py
@@ -20,8 +20,8 @@ for i in range(22):
     time_tgtcls.append(time_tgtcls[-1] + time_tgtcls[-2])
 
 dim_embeddings = [
-    ('origin_call', data.origin_call_size+1, 10),
-    ('origin_stand', data.stands_size+1, 10),
+    ('origin_call', data.origin_call_size, 10),
+    ('origin_stand', data.stands_size, 10),
     ('week_of_year', 52, 10),
     ('day_of_week', 7, 10),
     ('qhour_of_day', 24 * 4, 10),
diff --git a/config/joint_simple_mlp_tgtcls_1_cswdtx_bigger.py b/config/joint_simple_mlp_tgtcls_1_cswdtx_bigger.py
index 5548127..e96b44c 100644
--- a/config/joint_simple_mlp_tgtcls_1_cswdtx_bigger.py
+++ b/config/joint_simple_mlp_tgtcls_1_cswdtx_bigger.py
@@ -20,8 +20,8 @@ for i in range(21):
     time_tgtcls.append(time_tgtcls[-1] + time_tgtcls[-2])
 
 dim_embeddings = [
-    ('origin_call', data.origin_call_size+1, 15),
-    ('origin_stand', data.stands_size+1, 10),
+    ('origin_call', data.origin_call_size, 15),
+    ('origin_stand', data.stands_size, 10),
     ('week_of_year', 52, 10),
     ('day_of_week', 7, 10),
     ('qhour_of_day', 24 * 4, 10),
diff --git a/config/time_simple_mlp_2_cswdtx.py b/config/time_simple_mlp_2_cswdtx.py
index 0747e27..2ec28c6 100644
--- a/config/time_simple_mlp_2_cswdtx.py
+++ b/config/time_simple_mlp_2_cswdtx.py
@@ -9,8 +9,8 @@ n_begin_end_pts = 5     # how many points we consider at the beginning and end o
 n_valid = 1000
 
 dim_embeddings = [
-    ('origin_call', data.n_train_clients+1, 10),
-    ('origin_stand', data.n_stands+1, 10),
+    ('origin_call', data.origin_call_train_size, 10),
+    ('origin_stand', data.stands_size, 10),
     ('week_of_year', 52, 10),
     ('day_of_week', 7, 10),
     ('qhour_of_day', 24 * 4, 10),
diff --git a/config/time_simple_mlp_tgtcls_2_cswdtx.py b/config/time_simple_mlp_tgtcls_2_cswdtx.py
index d252e38..608ed7e 100644
--- a/config/time_simple_mlp_tgtcls_2_cswdtx.py
+++ b/config/time_simple_mlp_tgtcls_2_cswdtx.py
@@ -14,8 +14,8 @@ for i in range(22):
     tgtcls.append(tgtcls[-1] + tgtcls[-2])
 
 dim_embeddings = [
-    ('origin_call', data.origin_call_size+1, 10),
-    ('origin_stand', data.stands_size+1, 10),
+    ('origin_call', data.origin_call_size, 10),
+    ('origin_stand', data.stands_size, 10),
     ('week_of_year', 52, 10),
     ('day_of_week', 7, 10),
     ('qhour_of_day', 24 * 4, 10),