aboutsummaryrefslogtreecommitdiff
path: root/config/joint_simple_mlp_tgtcls_111_cswdtx_noise_dout.py
diff options
context:
space:
mode:
authorÉtienne Simon <esimon@esimon.eu>2015-05-22 11:23:49 -0400
committerÉtienne Simon <esimon@esimon.eu>2015-05-22 11:24:50 -0400
commit1137abf0511359df8f468f1d9a4828f3e0cb1b6a (patch)
tree76b0336d5ed3ee835b0efb902ab06ef4543a854d /config/joint_simple_mlp_tgtcls_111_cswdtx_noise_dout.py
parent39e549f05e568e4153381f025b3a0f256e9a7b7a (diff)
downloadtaxi-1137abf0511359df8f468f1d9a4828f3e0cb1b6a.tar.gz
taxi-1137abf0511359df8f468f1d9a4828f3e0cb1b6a.zip
Fix embeddings sizes.
Diffstat (limited to 'config/joint_simple_mlp_tgtcls_111_cswdtx_noise_dout.py')
-rw-r--r--config/joint_simple_mlp_tgtcls_111_cswdtx_noise_dout.py4
1 files changed, 2 insertions, 2 deletions
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),