diff options
| author | Étienne Simon <esimon@esimon.eu> | 2015-07-13 16:38:51 +0000 |
|---|---|---|
| committer | Étienne Simon <esimon@esimon.eu> | 2015-07-13 16:38:51 +0000 |
| commit | dc430951d6cb660ab804c7e6250aea1acc2dcd9d (patch) | |
| tree | 657aa861be60c1a88ec3cc69e4845cc18bd3b290 | |
| parent | 3fa43e0a437fa776e1dcb4949e7c6d7574239caf (diff) | |
| download | taxi-dc430951d6cb660ab804c7e6250aea1acc2dcd9d.tar.gz taxi-dc430951d6cb660ab804c7e6250aea1acc2dcd9d.zip | |
Fix at_least_k
| -rw-r--r-- | data/transformers.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/data/transformers.py b/data/transformers.py index f994488..6d3f488 100644 --- a/data/transformers.py +++ b/data/transformers.py @@ -14,7 +14,7 @@ fuel.config.default_seed = 123 def at_least_k(k, v, pad_at_begin, is_longitude): if len(v) == 0: - v = numpy.array([data.porto_center[1 if is_longitude else 0]], dtype=theano.config.floatX) + v = numpy.array([data.train_gps_mean[1 if is_longitude else 0]], dtype=theano.config.floatX) if len(v) < k: if pad_at_begin: v = numpy.concatenate((numpy.full((k - len(v),), v[0]), v)) |