Struct InMemoryDataset 

pub struct InMemoryDataset { /* private fields */ }

In-memory dataset with arrays.

Implementations

impl InMemoryDataset

pub fn new( inputs: Vec<Box<dyn ArrayProtocol>>, targets: Vec<Box<dyn ArrayProtocol>>, inputshape: Vec<usize>, outputshape: Vec<usize>, ) -> Self

Create a new in-memory dataset.

pub fn from_arrays<T, D1, D2>( inputs: Array<T, D1>, targets: Array<T, D2>, ) -> Self

where T: Clone + Send + Sync + 'static, D1: Dimension + Send + Sync, D2: Dimension + Send + Sync,

Create an in-memory dataset from arrays.

Trait Implementations

impl Dataset for InMemoryDataset

fn len(&self) -> usize

Get the number of samples in the dataset.

fn get( &self, index: usize, ) -> Option<(Box<dyn ArrayProtocol>, Box<dyn ArrayProtocol>)>

Get a sample from the dataset by index.

fn inputshape(&self) -> Vec<usize>

Get the input shape of the dataset.

fn outputshape(&self) -> Vec<usize>

Get the output shape of the dataset.

fn is_empty(&self) -> bool

Check if the dataset is empty.