diarization#

(s3prl.dataio.dataset.diarization)

Dataset#

class s3prl.dataio.dataset.diarization.Dataset(*args, **kwds)[source][source]#

Bases: Dataset

getinfo(index: int)[source][source]#

DiarizationDataset#

class s3prl.dataio.dataset.diarization.DiarizationDataset(mode, data_dir, chunk_size=2000, frame_shift=256, subsampling=1, rate=16000, use_last_samples=True, label_delay=0, num_speakers=None)[source][source]#

Bases: Dataset

getinfo(i)[source][source]#

KaldiData#

class s3prl.dataio.dataset.diarization.KaldiData(data_dir)[source][source]#

Bases: object

This class holds data in kaldi-style directory.

load_wav(recid, start=0, end=None)[source][source]#: Load wavfile given recid, start time and end time.