Struct ClassificationDataset

pub struct ClassificationDataset { /* private fields */ }

A dataset to train a Classifier.

Implementations

impl ClassificationDataset

pub fn builder<C: Class>() -> ClassificationDatasetBuilder<C>

Create a builder for a classification dataset.

pub fn save<P: AsRef<Path>>(&self, path: P) -> Result<()>

Save the dataset to the given path.

Example

let dev = candle_core::Device::Cpu;
let dataset = ClassificationDataset::load("dataset.safetensors", &dev).unwrap();
dataset.save("dataset_copy.safetensors").unwrap();

pub fn load<P: AsRef<Path>>(path: P, dev: &Device) -> Result<Self>

Load the dataset from the given path.

Example

let dev = candle_core::Device::Cpu;
let dataset = ClassificationDataset::load("dataset.safetensors", &dev).unwrap();

Trait Implementations

impl Clone for ClassificationDataset

fn clone(&self) -> ClassificationDataset

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for ClassificationDataset

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.