Struct SyntheticSentimentDataset 

pub struct SyntheticSentimentDataset { /* private fields */ }

A synthetic sentiment analysis dataset for testing.

Implementations

impl SyntheticSentimentDataset

pub fn new(size: usize, max_length: usize, vocab_size: usize) -> Self

Creates a new synthetic sentiment dataset.

pub fn small() -> Self

Creates a small test dataset.

pub fn train() -> Self

Creates a standard training dataset.

pub fn test() -> Self

Creates a standard test dataset.

Trait Implementations

impl Dataset for SyntheticSentimentDataset

type Item = (Tensor<f32>, Tensor<f32>)

The type of items in the dataset.

fn len(&self) -> usize

Returns the number of items in the dataset.

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

Gets an item by index.

fn is_empty(&self) -> bool

Returns true if the dataset is empty.