changing active learning definition plus main loops of mle/vi_blocks

1 files changed, 7 insertions, 6 deletions

diff --git a/simulation/mle_blocks.py b/simulation/mle_blocks.py
index ab8816f..0d27869 100644
--- a/simulation/mle_blocks.py
+++ b/simulation/mle_blocks.py
@@ -30,8 +30,9 @@ def create_mle_model(graph):
 
 if __name__ == "__main__":
     batch_size = 100
-    n_obs = 100000
-    graph = utils.create_wheel(100)
+    #n_obs = 100000
+    freq = 10
+    graph = utils.create_wheel(1000)
 
     print('GRAPH:\n', graph, '\n-------------\n')
 
@@ -51,10 +52,10 @@ if __name__ == "__main__":
         extensions=[
             be.FinishAfter(after_n_batches=10**3),
             bm.TrainingDataMonitoring([cost, params,
-                                       rmse, error], every_n_batches=10),
-            be.Printing(every_n_batches=10),
-            ub.JSONDump("log.json", every_n_batches=10),
-            ub.ActiveLearning(data_stream.dataset),
+                                       rmse, error], every_n_batches=freq),
+            be.Printing(every_n_batches=freq),
+            ub.JSONDump("logs/active_outdegree_mle.json", every_n_batches=freq),
+            ub.ActiveLearning(data_stream.dataset, graph, every_n_batches=freq),
         ],
     )
     loop.run()