diff options
| author | Thibaut Horel <thibaut.horel@gmail.com> | 2015-12-02 15:04:43 -0500 |
|---|---|---|
| committer | Thibaut Horel <thibaut.horel@gmail.com> | 2015-12-02 15:04:43 -0500 |
| commit | ccb192c4190701531094b46df85725158d4e9ffc (patch) | |
| tree | 31750bbaf1e6785cdb260cb5a2ae22f9f7cf8338 /simulation/vi_blocks.py | |
| parent | 5e546cb6c96e5e5e575730e27c175f558da5ec82 (diff) | |
| download | cascades-ccb192c4190701531094b46df85725158d4e9ffc.tar.gz | |
Fix the mess and use sqlite backend
Diffstat (limited to 'simulation/vi_blocks.py')
| -rw-r--r-- | simulation/vi_blocks.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/simulation/vi_blocks.py b/simulation/vi_blocks.py index 1177979..b78375b 100644 --- a/simulation/vi_blocks.py +++ b/simulation/vi_blocks.py @@ -51,7 +51,7 @@ def create_vi_model(n_nodes, n_samp=100): if __name__ == "__main__": - #n_cascades = 10000 + n_cascades = 10000 batch_size = 10 n_samples = 50 graph = utils.create_random_graph(n_nodes=4) @@ -70,6 +70,7 @@ if __name__ == "__main__": # data_stream = ub.dynamic_data_stream(graph, batch_size) loop = main_loop.MainLoop( alg, data_stream, + log_backend="sqlite", extensions=[ be.FinishAfter(after_n_batches=10**4), bm.TrainingDataMonitoring([cost, mu, sig, rmse], |