aboutsummaryrefslogtreecommitdiff
path: root/config/joint_simple_mlp_tgtcls_1_cswdtx.py
diff options
context:
space:
mode:
Diffstat (limited to 'config/joint_simple_mlp_tgtcls_1_cswdtx.py')
-rw-r--r--config/joint_simple_mlp_tgtcls_1_cswdtx.py7
1 files changed, 4 insertions, 3 deletions
diff --git a/config/joint_simple_mlp_tgtcls_1_cswdtx.py b/config/joint_simple_mlp_tgtcls_1_cswdtx.py
index 74b3c75..5da4130 100644
--- a/config/joint_simple_mlp_tgtcls_1_cswdtx.py
+++ b/config/joint_simple_mlp_tgtcls_1_cswdtx.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 = 5 # how many points we consider at the beginning and end of the known trajectory
n_end_pts = 5
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