Struct LabelEncoder 

pub struct LabelEncoder {
    pub classes: Vec<String>,
    /* private fields */
}

Encodes class labels as integers and decodes them back.

Fields

classes: Vec<String>

Class names in order (index 0 = first class).

Implementations

impl LabelEncoder

pub fn new() -> Self

Create an empty encoder.

pub fn fit(class_names: Vec<String>) -> Self

Fit the encoder from a list of class names.

pub fn encode(&self, name: &str) -> Option<usize>

Encode a class name to its index.

pub fn decode(&self, idx: usize) -> Option<&str>

Decode an index to its class name.

pub fn num_classes(&self) -> usize

Number of classes.

pub fn one_hot(&self, idx: usize) -> Vec<f64>

One-hot encode an index.

Trait Implementations

impl Clone for LabelEncoder

fn clone(&self) -> LabelEncoder

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for LabelEncoder

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

Formats the value using the given formatter.

impl Default for LabelEncoder

fn default() -> LabelEncoder

Returns the “default value” for a type.