diff options
| author | Alex Auvolat <alex.auvolat@ens.fr> | 2015-05-21 10:46:05 -0400 |
|---|---|---|
| committer | Alex Auvolat <alex.auvolat@ens.fr> | 2015-05-21 10:46:05 -0400 |
| commit | 9a779f7328a712a20dd393bdf32c6a84bf9fbe52 (patch) | |
| tree | 1906e27febd8b0b76ff1de4bdec74e56cbac15e4 /model/joint_simple_mlp_tgtcls.py | |
| parent | 9ff3d163609707c0138c0de731eec40449bd1815 (diff) | |
| download | taxi-9a779f7328a712a20dd393bdf32c6a84bf9fbe52.tar.gz taxi-9a779f7328a712a20dd393bdf32c6a84bf9fbe52.zip | |
Model changes
Diffstat (limited to 'model/joint_simple_mlp_tgtcls.py')
| -rw-r--r-- | model/joint_simple_mlp_tgtcls.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/model/joint_simple_mlp_tgtcls.py b/model/joint_simple_mlp_tgtcls.py index 834afbf..0aaf554 100644 --- a/model/joint_simple_mlp_tgtcls.py +++ b/model/joint_simple_mlp_tgtcls.py @@ -58,10 +58,12 @@ class Model(object): hidden = common_mlp.apply(inputs) dest_cls_probas = dest_mlp.apply(hidden) + # dest_cls_probas = theano.printing.Print("dest_cls_probas")(dest_cls_probas) dest_outputs = tensor.dot(dest_cls_probas, dest_classes) dest_outputs.name = 'dest_outputs' time_cls_probas = time_mlp.apply(hidden) + # time_cls_probas = theano.printing.Print("time_cls_probas")(time_cls_probas) time_outputs = tensor.dot(time_cls_probas, time_classes) + x_input_time time_outputs.name = 'time_outputs' |