diff options
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 |
commit | 60aa0d3fdd42d7489cc69acbb54c59d7c249ea34 (patch) | |
tree | 3ef37dab66250972cb226bff2ffb49f716a85915 /model | |
parent | c29a0d3f22134a8d1f5d557b325f6779c5961546 (diff) | |
download | taxi-60aa0d3fdd42d7489cc69acbb54c59d7c249ea34.tar.gz taxi-60aa0d3fdd42d7489cc69acbb54c59d7c249ea34.zip |
Adapt Alexandre's model to the new interface
Diffstat (limited to 'model')
-rw-r--r-- | model/dest_simple_mlp_tgtcls_alexandre.py | 8 |
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 = [] |