diff options
author | Étienne Simon <esimon@esimon.eu> | 2015-07-21 21:08:35 -0400 |
---|---|---|
committer | Étienne Simon <esimon@esimon.eu> | 2015-07-21 21:08:35 -0400 |
commit | 9013799ebca1c426c3c3e9019eb71018b253b025 (patch) | |
tree | b6cdaa6052478ce13204b16b5f42662cd69bd456 /model/bidirectional_direct.py | |
parent | e1673538607a7c8d784013b21b753f0c05c4cc34 (diff) | |
download | taxi-9013799ebca1c426c3c3e9019eb71018b253b025.tar.gz taxi-9013799ebca1c426c3c3e9019eb71018b253b025.zip |
Add bidirectional
Diffstat (limited to 'model/bidirectional_direct.py')
-rw-r--r-- | model/bidirectional_direct.py | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/model/bidirectional_direct.py b/model/bidirectional_direct.py new file mode 100644 index 0000000..81b33bd --- /dev/null +++ b/model/bidirectional_direct.py @@ -0,0 +1,13 @@ +from blocks.bricks.base import lazy + +from model.bidirectional import BidiRNN, Stream +import data + + +class Model(BidiRNN): + @lazy() + def __init__(self, config, **kwargs): + super(Model, self).__init__(config, **kwargs) + + def process_outputs(self, outputs): + return (outputs * data.train_gps_std) + data.train_gps_mean |