Struct burn_dataset::InMemDataset

pub struct InMemDataset<I> { /* private fields */ }

Dataset where all items are stored in ram.

Implementations

impl<I> InMemDataset<I>

pub fn new(items: Vec<I>) -> Self

Creates a new in memory dataset from the given items.

impl<I> InMemDataset<I>

where I: Clone + DeserializeOwned,

pub fn from_dataset(dataset: &impl Dataset<I>) -> Self

Create from a dataset. All items are loaded in memory.

pub fn from_json_rows<P: AsRef<Path>>(path: P) -> Result<Self, Error>

Create from a json rows file (one json per line).

Supported field types

pub fn from_csv<P: AsRef<Path>>( path: P, builder: &ReaderBuilder ) -> Result<Self, Error>

Create from a csv file.

The provided csv::ReaderBuilder can be configured to fit your csv format.

The supported field types are: String, integer, float, and bool.

See:

• Reading with Serde
• Delimiters, quotes and variable length records

Trait Implementations

impl<I> Dataset<I> for InMemDataset<I>

where I: Clone + Send + Sync,

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

Gets the item at the given index.

fn len(&self) -> usize

Gets the number of items in the dataset.

fn is_empty(&self) -> bool

Checks if the dataset is empty.

fn iter(&self) -> DatasetIterator<'_, I>

where Self: Sized,

Returns an iterator over the dataset.