diff options
Diffstat (limited to 'simulation/vi_blocks.py')
| -rw-r--r-- | simulation/vi_blocks.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/simulation/vi_blocks.py b/simulation/vi_blocks.py index e2e3bd9..0e671e4 100644 --- a/simulation/vi_blocks.py +++ b/simulation/vi_blocks.py @@ -77,9 +77,9 @@ if __name__ == "__main__": bm.TrainingDataMonitoring([cost, rmse, mu, error], every_n_batches=freq), be.Printing(every_n_batches=freq, after_epoch=False), - ub.JSONDump("logs/nonactive_vi.json", every_n_batches=freq), - # ub.ActiveLearning(dataset=data_stream.dataset, params=graph, - # every_n_batches=freq) + ub.JSONDump("logs/tmp.json", every_n_batches=freq), + ub.ActiveLearning(dataset=data_stream.dataset, params=sig, + every_n_batches=freq) ] ) loop.run() |