lumen-dataset 0.6.1

A tiny ML framework
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28

use std::convert::Infallible;
use crate::Dataset;

pub struct VecDataset<I> {
    items: Vec<I>,
}

impl<I> VecDataset<I> {
    pub fn new(items: Vec<I>) -> Self {
        Self { items }
    }
}

impl<I> Dataset for VecDataset<I> 
where 
    I: Clone
{
    type Item = I;
    type Error = Infallible;
    
    fn get(&self, index: usize) -> Result<Option<I>, Self::Error> {
        Ok(self.items.get(index).cloned())
    }

    fn len(&self) -> usize {
        self.items.len()
    }
}