+info about SpectrogramDatasetWithLength

argilla-io · Jun 14, 2017 · 09d06d7 · 09d06d7
1 parent b0370d0
commit 09d06d7
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/data/bucketing_sampler.py b/data/bucketing_sampler.py
@@ -6,6 +6,11 @@
 
 class SpectrogramDatasetWithLength(SpectrogramDataset):
  def __init__(self, *args, **kwargs):
+ """
+ SpectrogramDataset that splits utterances into buckets based on their length.
+ Bucketing is done via numpy's histogram method.
+ Used by BucketingSampler to sample utterances from the same bin.
+ """
  super(SpectrogramDatasetWithLength, self).__init__(*args, **kwargs)
  audio_paths = [path for (path, _) in self.ids]
  audio_lengths = [len(load_audio(path)) for path in audio_paths]