update exmaple

erdogant · Dec 18, 2021 · 47f4de6 · 47f4de6
1 parent 76ff966
commit 47f4de6
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/bnlearn/examples.py b/bnlearn/examples.py
@@ -284,12 +284,13 @@
 
 df = bn.import_example('asia')
 edges = [('smoke', 'lung'),
-         ('smoke', 'bronc'),
-         ('lung', 'xray'),
-         ('bronc', 'xray')]
+          ('smoke', 'bronc'),
+          ('lung', 'xray'),
+          ('bronc', 'xray')]
+
 
 # Make the actual Bayesian DAG
-DAG = bn.make_DAG(edges, verbose=0)
+DAG = bn.make_DAG(edges, verbose=0, methodtype='bayes')
 model = bn.parameter_learning.fit(DAG, df, verbose=3)
 # Generate some data based on DAG
 df = bn.sampling(model, n=1000)