pgmpy.datasets.load_dataset

pgmpy.datasets.load_dataset(name: str) → Dataset

Load a dataset by name.

Parameters:

namestr

Name of the dataset to load.

Examples

>>> from pgmpy.datasets import load_dataset
>>> dataset = load_dataset("sachs_mixed")
>>> df = dataset.data
>>> ground_truth = dataset.ground_truth