aboutsummaryrefslogtreecommitdiff
path: root/config/simple_mlp_2_cs.py
diff options
context:
space:
mode:
authorAlex Auvolat <alex.auvolat@ens.fr>2015-05-05 14:15:21 -0400
committerAlex Auvolat <alex.auvolat@ens.fr>2015-05-05 14:15:21 -0400
commit54613c1f9cf510ca7a71d6619418f2247515aec6 (patch)
treebed9a5a11ef5b7feecee44095a29400e32f76b05 /config/simple_mlp_2_cs.py
parent712035b88be1816d3fbd58ce69ae6464767c780e (diff)
downloadtaxi-54613c1f9cf510ca7a71d6619418f2247515aec6.tar.gz
taxi-54613c1f9cf510ca7a71d6619418f2247515aec6.zip
Add models for time predictioAdd models for time prediction
Diffstat (limited to 'config/simple_mlp_2_cs.py')
-rw-r--r--config/simple_mlp_2_cs.py21
1 files changed, 0 insertions, 21 deletions
diff --git a/config/simple_mlp_2_cs.py b/config/simple_mlp_2_cs.py
deleted file mode 100644
index fa2f4c1..0000000
--- a/config/simple_mlp_2_cs.py
+++ /dev/null
@@ -1,21 +0,0 @@
-import model.simple_mlp as model
-
-import data
-
-n_begin_end_pts = 5 # how many points we consider at the beginning and end of the known trajectory
-n_end_pts = 5
-
-n_valid = 1000
-
-dim_embeddings = [
- ('origin_call', data.n_train_clients+1, 10),
- ('origin_stand', data.n_stands+1, 10)
-]
-
-dim_input = n_begin_end_pts * 2 * 2 + sum(x for (_, _, x) in dim_embeddings)
-dim_hidden = [200, 100]
-dim_output = 2
-
-learning_rate = 0.0001
-momentum = 0.99
-batch_size = 32