aboutsummaryrefslogtreecommitdiff
path: root/config/joint_simple_mlp_tgtcls_111_cswdtx_bigger.py
diff options
context:
space:
mode:
authorAlex Auvolat <alex.auvolat@ens.fr>2015-05-21 11:01:00 -0400
committerAlex Auvolat <alex.auvolat@ens.fr>2015-05-21 11:01:00 -0400
commitf6d2c6fc47f93b158b70b5c0c9a45324041ca4d5 (patch)
tree69ec2b5d0d3b949c4ed19e1c287a43f5db4c8929 /config/joint_simple_mlp_tgtcls_111_cswdtx_bigger.py
parentb956078f7be28ea9dc763479a572f9e2efbeabaa (diff)
downloadtaxi-f6d2c6fc47f93b158b70b5c0c9a45324041ca4d5.tar.gz
taxi-f6d2c6fc47f93b158b70b5c0c9a45324041ca4d5.zip
Fix new models.
Diffstat (limited to 'config/joint_simple_mlp_tgtcls_111_cswdtx_bigger.py')
-rw-r--r--config/joint_simple_mlp_tgtcls_111_cswdtx_bigger.py7
1 files changed, 4 insertions, 3 deletions
diff --git a/config/joint_simple_mlp_tgtcls_111_cswdtx_bigger.py b/config/joint_simple_mlp_tgtcls_111_cswdtx_bigger.py
index 8adb6e7..4432b5d 100644
--- a/config/joint_simple_mlp_tgtcls_111_cswdtx_bigger.py
+++ b/config/joint_simple_mlp_tgtcls_111_cswdtx_bigger.py
@@ -1,17 +1,18 @@
+import os
import cPickle
-import model.joint_simple_mlp_tgtcls as model
-
from blocks.initialization import IsotropicGaussian, Constant
import data
+from model.joint_simple_mlp_tgtcls import Model, Stream
+
n_begin_end_pts = 10 # how many points we consider at the beginning and end of the known trajectory
n_end_pts = 10
n_valid = 1000
-with open("%s/arrival-clusters.pkl" % data.path) as f:
+with open(os.path.join(data.path, 'arrival-clusters.pkl')) as f:
dest_tgtcls = cPickle.load(f)
# generate target classes for time prediction as a Fibonacci sequence