asteroid.data.wsj0_mix module

asteroid.data.wsj0_mix.make_dataloaders(train_dir, valid_dir, n_src=2, sample_rate=8000, segment=4.0, batch_size=4, num_workers=None, **kwargs)

class asteroid.data.wsj0_mix.Wsj0mixDataset(json_dir, n_src=2, sample_rate=8000, segment=4.0)

Bases: sphinx.ext.autodoc.importer._MockObject

Dataset class for the wsj0-mix source separation dataset.

Parameters:

• json_dir (str) – The path to the directory containing the json files.
• sample_rate (int, optional) – The sampling rate of the wav files.
• segment (float, optional) – Length of the segments used for training, in seconds. If None, use full utterances (e.g. for test).
• n_src (int, optional) – Number of sources in the training targets.

References

“Deep clustering: Discriminative embeddings for segmentation and separation”, Hershey et al. 2015.

dataset_name = 'wsj0-mix'

__getitem__(idx)

Gets a mixture/sources pair. :returns: mixture, vstack([source_arrays])

get_infos()

Get dataset infos (for publishing models).

Returns:dict, dataset infos with keys dataset, task and licences.