From ccb192c4190701531094b46df85725158d4e9ffc Mon Sep 17 00:00:00 2001
From: Thibaut Horel <thibaut.horel@gmail.com>
Date: Wed, 2 Dec 2015 15:04:43 -0500
Subject: Fix the mess and use sqlite backend

---
 simulation/vi_blocks.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'simulation/vi_blocks.py')

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],
-- 
cgit v1.2.3-70-g09d2