mit_d3m package

Submodules

• mit_d3m.config module
• mit_d3m.dataset module
• mit_d3m.db module
• mit_d3m.loaders module
• mit_d3m.metrics module
• mit_d3m.stats module
• mit_d3m.utils module

Module contents

Top-level package for mit-d3m.

Functions

load_d3mds(dataset[, root, force_download])	Load dataset into D3MDS format, as necessary downloading tarfile from S3 and extracting
load_dataset(dataset[, root, force_download])	Load dataset, as necessary downloading tarfile from S3 and extracting

mit_d3m.load_d3mds(dataset, root='data', force_download=False)

Load dataset into D3MDS format, as necessary downloading tarfile from S3 and extracting

If the root directory is

Parameters

• dataset (str) – dataset identifier

• root (path-like, optional) – root directory to store tarfiles and extracted datasets. Defaults to ‘./data/’.

• force_download (boolean, optional) – download the tarfile even if it already exists, also causing the files to be re-extracted. Defaults to False.

Returns

mit_d3m.dataset.D3MDS

mit_d3m.load_dataset(dataset, root='data', force_download=False)

Load dataset, as necessary downloading tarfile from S3 and extracting

Parameters

• dataset (str) – dataset identifier

• root (path-like, optional) – root directory to store tarfiles and extracted datasets. Defaults to ‘./data/’.

• force_download (boolean, optional) – download the tarfile even if it already exists, also causing the files to be re-extracted. Defaults to False.

Returns

mit_d3m.loaders.Dataset