Adapting examples/tutorials

sbi-dev · Jul 24, 2020 · 7a5a625 · 7a5a625
1 parent a2db5df
commit 7a5a625
Show file tree

Hide file tree

Showing 5 changed files with 196 additions and 51 deletions.
diff --git a/sbi/examples/minimal.py b/sbi/examples/minimal.py
@@ -1,5 +1,6 @@
 import torch
-from sbi.inference import infer, prepare_for_sbi, SNPE
+
+from sbi.inference import SNPE, infer, prepare_for_sbi
 from sbi.simulators.linear_gaussian import diagonal_linear_gaussian
 
 
@@ -33,7 +34,7 @@ def flexible():
     )
     simulator, prior = prepare_for_sbi(diagonal_linear_gaussian, prior)
     inference = SNPE(simulator, prior)
-    posterior = inference(num_rounds=1, num_simulations_per_round=500)
+    posterior = inference(num_simulations=500)
     posterior.sample((100,), x=x_o)
 
     return posterior

diff --git a/tutorial/00_getting_started.ipynb b/tutorial/00_getting_started.ipynb