Struct ClassificationDatasetBuilder

pub struct ClassificationDatasetBuilder<C: Class> { /* private fields */ }

A builder for ClassificationDataset.

Implementations

impl<C: Class> ClassificationDatasetBuilder<C>

pub fn new() -> Self

Create a new dataset builder.

pub fn add(&mut self, input: impl Into<Box<[f32]>>, class: C)

Adds a pair of input and class to the dataset.

Example

let mut dataset = ClassificationDatasetBuilder::new();
dataset.add(vec![1.0, 2.0, 3.0, 4.0], MyClass::Person);
dataset.add(vec![4.0, 3.0, 2.0, 1.0], MyClass::Thing);

pub fn build(self, dev: &Device) -> Result<ClassificationDataset>

Builds the dataset and copies the data to the device passed in.

Example

let dev = candle_core::Device::Cpu;
let mut dataset = ClassificationDatasetBuilder::new();
dataset.add(vec![1.0, 2.0, 3.0, 4.0], MyClass::Person);
dataset.add(vec![4.0, 3.0, 2.0, 1.0], MyClass::Thing);
let dataset = dataset.build(&dev).unwrap();

Trait Implementations

impl<C: Class> Default for ClassificationDatasetBuilder<C>

fn default() -> Self

Returns the “default value” for a type.