From 1137abf0511359df8f468f1d9a4828f3e0cb1b6a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C3=89tienne=20Simon?= <esimon@esimon.eu>
Date: Fri, 22 May 2015 11:23:49 -0400
Subject: Fix embeddings sizes.

---
 config/time_simple_mlp_2_cswdtx.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'config/time_simple_mlp_2_cswdtx.py')

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),
-- 
cgit v1.2.3