diff options
| author | Alex Auvolat <alex.auvolat@ens.fr> | 2015-07-23 10:13:28 -0400 |
|---|---|---|
| committer | Alex Auvolat <alex.auvolat@ens.fr> | 2015-07-23 10:13:28 -0400 |
| commit | 9313c8bb7f62b14f8c15d2119ae678641f751dbd (patch) | |
| tree | 646be5562b40d1203fc32764f956c03e04c99dde /model/mlp.py | |
| parent | dd8ae5ea8ed0c7cb1a7880b1e1887c6e23cdf910 (diff) | |
| download | taxi-9313c8bb7f62b14f8c15d2119ae678641f751dbd.tar.gz taxi-9313c8bb7f62b14f8c15d2119ae678641f751dbd.zip | |
Can do test with batches of size >1
Diffstat (limited to 'model/mlp.py')
| -rw-r--r-- | model/mlp.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/model/mlp.py b/model/mlp.py index 7d04c82..d24b2cc 100644 --- a/model/mlp.py +++ b/model/mlp.py @@ -106,7 +106,7 @@ class Stream(object): stream = transformers.taxi_add_first_last_len(stream, self.config.n_begin_end_pts) stream = transformers.taxi_remove_test_only_clients(stream) - return Batch(stream, iteration_scheme=ConstantScheme(1)) + return Batch(stream, iteration_scheme=ConstantScheme(self.config.batch_size)) def inputs(self): return {'call_type': tensor.bvector('call_type'), |