pymc-devs / pymc3 / 3790

87%

push

travis-ci

Generators as Data and logp rescaling (#1684)

* Added mode argument to several step methods and advi to allow mode setting (e.g. nanguardmode) for Theano functions

* Created Generator Op with simple Test

* added ndim test

* updated test

* updated test, added test value check

* added test for replacing generator with shared variable

* added shortcut for generator op

* refactored test

* added population kwarg (no tests yet)

* added population kwarg for free var(autoencoder case)

* Revert "Added mode argument to several step methods and advi to allow mode setting (e.g. nanguardmode) for Theano functions"

This reverts commit 6bdb53a06.

* add docstring to generator Op

* rename population -> total_size

* update docstrings in model

* fix typo in `as_tensor` function

* add test for density rescaling

* add test for density with generators

* fix python2 error with generators

* test nan produced instead of error when generator is exhausted

* Setting new generator is now available

* Refactor test

* Refactor assertions

* fix typo

* refactor casting to DataGenerator

* make pickling potentially possible

* refactor logp scaling

* better resolving generator types with python2 and python3

192 of 192 new or added lines in 7 files covered. (100.0%)

7503 of 8853 relevant lines covered (84.75%)

2.74 hits per line