Struct burn::data::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>) -> InMemDataset<I>

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>) -> InMemDataset<I>

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

pub fn from_json_rows<P>(path: P) -> Result<InMemDataset<I>, Error>

where P: AsRef<Path>,

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

Supported field types

pub fn from_csv<P>( path: P, builder: &ReaderBuilder, ) -> Result<InMemDataset<I>, Error>

where P: AsRef<Path>,

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.

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

where Self: Sized,

Returns a new Dataset of all the windows of length size. The windows overlap. Is empty if the input Dataset is shorter than size.