diff options
| author | Étienne Simon <esimon@esimon.eu> | 2015-07-21 18:26:43 -0400 |
|---|---|---|
| committer | Étienne Simon <esimon@esimon.eu> | 2015-07-21 18:27:55 -0400 |
| commit | e1673538607a7c8d784013b21b753f0c05c4cc34 (patch) | |
| tree | f42e316e0c5bf67e3c9953aad6ba8fe9656829f2 /model/rnn_direct.py | |
| parent | 58dcf7b17e9db6af53808994a7d39a759fcc5028 (diff) | |
| download | taxi-e1673538607a7c8d784013b21b753f0c05c4cc34.tar.gz taxi-e1673538607a7c8d784013b21b753f0c05c4cc34.zip | |
Genericize RNNs
Diffstat (limited to 'model/rnn_direct.py')
| -rw-r--r-- | model/rnn_direct.py | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/model/rnn_direct.py b/model/rnn_direct.py new file mode 100644 index 0000000..8123327 --- /dev/null +++ b/model/rnn_direct.py @@ -0,0 +1,13 @@ +from blocks.bricks.base import lazy + +from model.rnn import RNN, Stream +import data + + +class Model(RNN): + @lazy() + def __init__(self, config, **kwargs): + super(Model, self).__init__(config, **kwargs) + + def process_rto(self, rto): + return (rto * data.train_gps_std) + data.train_gps_mean |