model/rnn_direct.py


1
2
3
4
5
6
7
8
9
10
11
12
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