Struct AudioClassificationDataset 

pub struct AudioClassificationDataset { /* private fields */ }

A dataset for audio classification tasks.

Implementations

impl AudioClassificationDataset

pub fn new( waveforms: Vec<Tensor<f32>>, labels: Vec<usize>, sample_rate: usize, num_classes: usize, ) -> Self

Creates a new audio classification dataset from waveforms and labels.

pub fn sample_rate(&self) -> usize

Returns the sample rate.

pub fn num_classes(&self) -> usize

Returns the number of classes.

Trait Implementations

impl Dataset for AudioClassificationDataset

type Item = (Tensor<f32>, Tensor<f32>)

The type of items in the dataset.

fn len(&self) -> usize

Returns the number of items in the dataset.

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

Gets an item by index.

fn is_empty(&self) -> bool

Returns true if the dataset is empty.