update examples discritizer

erdogant · Oct 7, 2024 · 761d514 · 761d514
1 parent e8a997d
commit 761d514
Showing 1 changed file with 5 additions and 13 deletions.
diff --git a/bnlearn/examples_discretize.py b/bnlearn/examples_discretize.py
@@ -2,19 +2,9 @@
 import pandas as pd
 import bnlearn as bn
 
-df = pd.read_csv(
-    "bnlearn/tests/discretize/data/auto_mpg.csv",
-    dtype={
-        "mpg": "float64",
-        "cylinders": "int64",
-        "displacement": "float64",
-        "horsepower": "float64",
-        "weight": "float64",
-        "acceleration": "float64",
-        "model_year": "int64",
-        "origin": "int64",
-    },
-)
+#%%
+df = bn.import_example('auto_mpg')
+del df['origin']
 
 edges = [
     ("cylinders", "displacement"),
@@ -74,3 +64,5 @@
 evidence = {"weight": bn.discretize_value(df_disc["weight"], 3000.0)}
 print(evidence)
 print(bn.inference.fit(model_mle, variables=["mpg"], evidence=evidence, verbose=0))
+
+# %%