Skip to main content

SpeechCommandsDataset

Struct SpeechCommandsDataset 

Source
pub struct SpeechCommandsDataset { /* private fields */ }
Available on crate features std and dataset and audio only.
Expand description

Speech Commands dataset from Huggingface v0.02. See Speech Commands dataset.

The data is downloaded from Huggingface and stored in a SQLite database (3.0 GB). The dataset contains 99,720 audio samples of 2,607 people saying 35 different words.

NOTE: The most samples are under 1 second long but there are some with pure background noise that need splitting into shorter segmants.

The labels are 20 target words, silence and other words.

The dataset is split into 3 parts:

  • train: 84,848 audio files
  • test: 4,890 audio files
  • validation: 9,982 audio files

Implementations§

Source§

impl SpeechCommandsDataset

Source

pub fn new(split: &str) -> SpeechCommandsDataset

Create a new dataset with the given split.

Source

pub fn train() -> SpeechCommandsDataset

Create a new dataset with the train split.

Source

pub fn test() -> SpeechCommandsDataset

Create a new dataset with the test split.

Source

pub fn validation() -> SpeechCommandsDataset

Create a new dataset with the validation split.

Source

pub fn num_classes() -> usize

Returns the number of classes in the dataset

Trait Implementations§

Source§

impl Dataset<SpeechItem> for SpeechCommandsDataset

Source§

fn get(&self, index: usize) -> Option<SpeechItem>

Gets the item at the given index.
Source§

fn len(&self) -> usize

Gets the number of items in the dataset.
Source§

fn is_empty(&self) -> bool

Checks if the dataset is empty.
Source§

fn iter(&self) -> DatasetIterator<'_, I>
where Self: Sized,

Returns an iterator over the dataset.

Auto Trait Implementations§

Blanket Implementations§

Source§

impl<T> Any for T
where T: 'static + ?Sized,

Source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
Source§

impl<T> Borrow<T> for T
where T: ?Sized,

Source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
Source§

impl<T> BorrowMut<T> for T
where T: ?Sized,

Source§

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more
Source§

impl<T> From<T> for T

Source§

fn from(t: T) -> T

Returns the argument unchanged.

Source§

impl<T, U> Into<U> for T
where U: From<T>,

Source§

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

Source§

impl<T, U> TryFrom<U> for T
where U: Into<T>,

Source§

type Error = Infallible

The type returned in the event of a conversion error.
Source§

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.
Source§

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

Source§

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.
Source§

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.
Source§

impl<V, T> VZip<V> for T
where V: MultiLane<T>,

Source§

fn vzip(self) -> V

Source§

impl<I, T> Window<I> for T
where T: Dataset<I> + ?Sized,

Source§

fn window(&self, current: usize, size: NonZero<usize>) -> Option<Vec<I>>

Creates a window of a collection. Read more
Source§

impl<I, T> Windows<I> for T
where T: Dataset<I>,

Source§

fn windows(&self, size: usize) -> WindowsIterator<'_, I>

Is empty if the Dataset is shorter than size.

§Panics

Panics if size is 0.

§Examples
use crate::burn_dataset::{
   transform::{Windows, WindowsDataset},
   Dataset, InMemDataset,
};

let items = [1, 2, 3, 4].to_vec();
let dataset = InMemDataset::new(items.clone());

for window in dataset.windows(2) {
 // do sth with window
}