aboutsummaryrefslogtreecommitdiff
path: root/config/model_0.py
diff options
context:
space:
mode:
authorAlex Auvolat <alex.auvolat@ens.fr>2015-04-29 19:09:37 -0400
committerAlex Auvolat <alex.auvolat@ens.fr>2015-04-29 19:09:37 -0400
commitfe704521e6cb4d7f32414b55044b6e2240524bf5 (patch)
treec6b02de77fc274a02f135ad6a71b586d0cb591a0 /config/model_0.py
parente28390de61b23882f6e1069d565a2137825c2662 (diff)
downloadtaxi-fe704521e6cb4d7f32414b55044b6e2240524bf5.tar.gz
taxi-fe704521e6cb4d7f32414b55044b6e2240524bf5.zip
Fix CSV import (partially)
Diffstat (limited to 'config/model_0.py')
-rw-r--r--config/model_0.py6
1 files changed, 1 insertions, 5 deletions
diff --git a/config/model_0.py b/config/model_0.py
index ba04a15..736ef30 100644
--- a/config/model_0.py
+++ b/config/model_0.py
@@ -2,16 +2,12 @@ n_dow = 7 # number of division for dayofweek/dayofmonth/hourofday
n_dom = 31
n_hour = 24
-n_clients = 57124 #57105
-n_stands = 63
-
n_begin_end_pts = 5 # how many points we consider at the beginning and end of the known trajectory
n_end_pts = 5
-train_size = 1710670
n_valid = 1000
-dim_embed = 50
+dim_embed = 10
dim_input = n_begin_end_pts * 2 * 2 + dim_embed + dim_embed
dim_hidden = [200]
dim_output = 2