Skip to content

windisch/bn_testing

Repository files navigation

BN testing

Test Package Documentation Status PyPI

A test framework to evaluate methods that learn Bayesian Networks from high-dimensional observational data.

Sampling

Set up the graphical model and sample data

from bn_testing.models import BayesianNetwork
from bn_testing.dags import ErdosReny
from bn_testing.conditionals import PolynomialConditional


model = BayesianNetwork(
   dag=ErdosReny(p=0.01, n_nodes=100),
   conditionals=PolynomialConditional(max_terms=5)
)

df = model.sample(10000, normalize=True)

The observations are stored in a pandas.DataFrame where the columns are the nodes of the DAG and each row is an observation. The underlying DAG of the graphical model can be accessed with model.dag

About

A test bench for graphical models

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages