Struct NumpyDataset 

pub struct NumpyDataset { /* private fields */ }

Dataset loaded from NumPy .npy files

Implementations

impl NumpyDataset

pub fn from_directory<P: AsRef<Path>>(path: P) -> Result<Self>

Load from directory containing .npy files and dataset.json

pub fn len(&self) -> usize

Get number of examples

pub fn is_empty(&self) -> bool

Check if empty

pub fn vocab_size(&self) -> usize

Get vocabulary size

pub fn seq_len(&self) -> usize

Get sequence length

pub fn metadata(&self) -> &DatasetMetadata

Get metadata

pub fn get_input(&self, idx: usize) -> ArrayView1<'_, i32>

Get input at index

pub fn get_label(&self, idx: usize) -> ArrayView1<'_, i32>

Get label at index

pub fn get_puzzle_id(&self, idx: usize) -> Option<i32>

Get puzzle ID at index (if available)