Struct tch::data::TextData

pub struct TextData { /* fields omitted */ }

Text data holder.

Implementations

impl TextData

pub fn new<P: AsRef<Path>>(filename: P) -> Result<TextData, TchError>

Creates a text dataset from a file.

pub fn labels(&self) -> i64

Returns the number of different characters/labels used by the dataset.

pub fn data(&self) -> Tensor

Returns a shallow copy of the data.

pub fn label_to_char(&self, label: i64) -> char

pub fn char_to_label(&self, c: char) -> Result<u8, TchError>

pub fn iter_shuffle(&self, seq_len: i64, batch_size: i64) -> TextDataIter

Returns a batch iterator over the dataset. Each sample is made of seq_len characters.

Trait Implementations

impl Debug for TextData

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.