aboutsummaryrefslogtreecommitdiff
path: root/model/dest_simple_mlp_tgtcls_alexandre.py
diff options
context:
space:
mode:
authorÉtienne Simon <esimon@esimon.eu>2015-05-05 23:03:13 -0400
committerÉtienne Simon <esimon@esimon.eu>2015-05-05 23:03:13 -0400
commit60aa0d3fdd42d7489cc69acbb54c59d7c249ea34 (patch)
tree3ef37dab66250972cb226bff2ffb49f716a85915 /model/dest_simple_mlp_tgtcls_alexandre.py
parentc29a0d3f22134a8d1f5d557b325f6779c5961546 (diff)
downloadtaxi-60aa0d3fdd42d7489cc69acbb54c59d7c249ea34.tar.gz
taxi-60aa0d3fdd42d7489cc69acbb54c59d7c249ea34.zip
Adapt Alexandre's model to the new interface
Diffstat (limited to 'model/dest_simple_mlp_tgtcls_alexandre.py')
-rw-r--r--model/dest_simple_mlp_tgtcls_alexandre.py8
1 files changed, 4 insertions, 4 deletions
diff --git a/model/dest_simple_mlp_tgtcls_alexandre.py b/model/dest_simple_mlp_tgtcls_alexandre.py
index 87e20a3..825bf80 100644
--- a/model/dest_simple_mlp_tgtcls_alexandre.py
+++ b/model/dest_simple_mlp_tgtcls_alexandre.py
@@ -14,11 +14,11 @@ import error
class Model(object):
def __init__(self, config):
# The input and the targets
- x_firstk_latitude = (tensor.matrix('first_k_latitude') - data.porto_center[0]) / data.data_std[0]
- x_firstk_longitude = (tensor.matrix('first_k_longitude') - data.porto_center[1]) / data.data_std[1]
+ x_firstk_latitude = (tensor.matrix('first_k_latitude') - data.train_gps_mean[0]) / data.train_gps_std[0]
+ x_firstk_longitude = (tensor.matrix('first_k_longitude') - data.train_gps_mean[1]) / data.train_gps_std[1]
- x_lastk_latitude = (tensor.matrix('last_k_latitude') - data.porto_center[0]) / data.data_std[0]
- x_lastk_longitude = (tensor.matrix('last_k_longitude') - data.porto_center[1]) / data.data_std[1]
+ x_lastk_latitude = (tensor.matrix('last_k_latitude') - data.train_gps_mean[0]) / data.train_gps_std[0]
+ x_lastk_longitude = (tensor.matrix('last_k_longitude') - data.train_gps_mean[1]) / data.train_gps_std[1]
input_list = [x_firstk_latitude, x_firstk_longitude, x_lastk_latitude, x_lastk_longitude]
embed_tables = []