Struct ParameterLearner 

pub struct ParameterLearner { /* private fields */ }

Implementations

impl ParameterLearner

pub fn new(alpha: f64, beta: f64, base_rate: Option<f64>) -> ScoringResult<Self>

pub fn alpha(&self) -> f64

pub fn beta(&self) -> f64

pub fn base_rate(&self) -> Option<f64>

pub fn params(&self) -> BTreeMap<String, f64>

pub fn probability(&self, raw_score: f64) -> ScoringResult<f64>

pub fn update( &mut self, raw_score: f64, label: f64, learning_rate: f64, ) -> ScoringResult<()>

Apply one exact logistic-loss gradient step to a raw BM25 score.

pub fn fit( &mut self, raw_scores: &[f64], labels: &[f64], learning_rate: f64, epochs: usize, ) -> ScoringResult<BTreeMap<String, f64>>

pub fn fit_with_options( &mut self, raw_scores: &[f64], labels: &[f64], ) -> ScoringResult<BTreeMap<String, f64>>

Trait Implementations

impl Clone for ParameterLearner

fn clone(&self) -> ParameterLearner

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ParameterLearner

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

Formats the value using the given formatter.

impl Default for ParameterLearner

fn default() -> Self

Returns the “default value” for a type.