Struct MultiLabelClassifier 

pub struct MultiLabelClassifier { /* private fields */ }

Multi-label classifier using binary relevance approach

Trains one binary classifier per label, allowing texts to belong to multiple categories.

Implementations

impl MultiLabelClassifier

pub fn new(threshold: f64) -> Self

Create a new multi-label classifier

pub fn fit( &mut self, features: &Array2<f64>, label_sets: &[Vec<String>], ) -> Result<()>

Train the classifier

Arguments

• features - Feature matrix (n_samples x n_features)
• label_sets - For each sample, a set of labels it belongs to

pub fn predict( &self, features: &Array2<f64>, ) -> Result<Vec<MultiLabelPrediction>>

Predict labels for feature matrix

Trait Implementations

impl Clone for MultiLabelClassifier

fn clone(&self) -> MultiLabelClassifier

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for MultiLabelClassifier

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

Formats the value using the given formatter.