Struct LRClassifier 

pub struct LRClassifier { /* private fields */ }

A classifier that uses a trained LR model for prediction.

Implementations

impl LRClassifier

pub fn new() -> Self

Create a new classifier with an empty model.

pub fn from_model(model: LRModel) -> Self

Create a classifier from an existing model.

pub fn load(path: String) -> Result<Self, String>

Load a model from a file.

pub fn model(&self) -> &LRModel

Get the underlying model.

pub fn predict(&self, features: &[String]) -> String

Predict the most likely class for the given features.

Returns the class label with the highest probability.

pub fn predict_with_prob(&self, features: &[String]) -> (String, f64)

Predict with probability for the most likely class.

Returns (class_label, probability).

pub fn predict_proba(&self, features: &[String]) -> Vec<(String, f64)>

Get probability distribution over all classes.

Returns a list of (class_label, probability) tuples, sorted by probability descending.

pub fn predict_top_k(&self, features: &[String], k: usize) -> Vec<(String, f64)>

Get top-k most likely classes with probabilities.

Returns a list of (class_label, probability) tuples for the top-k classes.

pub fn num_classes(&self) -> usize

Get the number of classes.

pub fn classes(&self) -> Vec<String>

Get all class labels.

Trait Implementations

impl Clone for LRClassifier

fn clone(&self) -> LRClassifier

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for LRClassifier

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

Formats the value using the given formatter.

impl Default for LRClassifier

fn default() -> Self

Returns the “default value” for a type.