rougers 0.2.1

Fast ROUGE metrics with scores identical to Google's rouge-score
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

/// Precision, recall and F1 for one ROUGE variant.
#[derive(Clone, Copy, Debug, Default, PartialEq)]
pub struct Score {
    pub precision: f64,
    pub recall: f64,
    pub fmeasure: f64,
}

impl Score {
    pub(crate) const ZERO: Score = Score {
        precision: 0.0,
        recall: 0.0,
        fmeasure: 0.0,
    };

    pub(crate) fn from_counts(hits: usize, prediction_len: usize, target_len: usize) -> Score {
        let precision = hits as f64 / prediction_len as f64;
        let recall = hits as f64 / target_len as f64;
        Score {
            precision,
            recall,
            fmeasure: fmeasure(precision, recall),
        }
    }
}

/// The harmonic mean of precision and recall, or zero when both are zero.
pub fn fmeasure(precision: f64, recall: f64) -> f64 {
    if precision + recall > 0.0 {
        2.0 * precision * recall / (precision + recall)
    } else {
        0.0
    }
}