inferlo.datasets.DatasetLoader

class inferlo.datasets.DatasetLoader[source]

Loads graphical models from named datasets.

__init__(data_dir=None)[source]

Parameters:: data_dir – Where to store cached datasets. Specify if you want datasets being cached locally. If not set, default system temporary directory will be used.

Methods

`__init__`([data_dir])
`custom_dataset_exists`(name)	Checks if custom dataset exists.
`load_custom_dataset`(name)	Loads previously saved datataset.
`load_file_`(url_prefix, file_name)	Loads file from the web to local file.
`load_uai_dataset`(dataset_name)	Loads named dataset from UAI competition.
`save_custom_dataset`(dataset)	Saves dataset locally.

custom_dataset_exists(name: str)[source]: Checks if custom dataset exists.

load_custom_dataset(name: str) → Dataset[source]: Loads previously saved datataset.

load_file_(url_prefix, file_name)[source]

Loads file from the web to local file.

If local file already exists, does nothing. Returns local path to loaded file.

load_uai_dataset(dataset_name) → Dataset[source]

Loads named dataset from UAI competition.

Parameters:: dataset_name – Name of dataset, e.g. “Promedus_11.uai”. For full list of UAI datasets, see http://sli.ics.uci.edu/~ihler/uai-data/. Not all of them are currently supported.
Returns:: Dataset object, containing graphical model (as GenericGraphModel object), true logarithm of the partition function and true marginals.