shap-rs 0.1.0

Native Rust implementations of model-agnostic, linear, and TreeSHAP explainers
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

use crate::{Explanation, Result, ShapError};
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]
pub struct WaterfallItem {
    pub feature: usize,
    pub value: f64,
    pub contribution: f64,
}
pub fn data(e: &Explanation, sample: usize, output: usize) -> Result<Vec<WaterfallItem>> {
    if sample >= e.n_samples() {
        return Err(ShapError::InvalidSampleIndex {
            index: sample,
            n_samples: e.n_samples(),
        });
    }
    if output >= e.n_outputs() {
        return Err(ShapError::InvalidOutputIndex {
            index: output,
            n_outputs: e.n_outputs(),
        });
    }
    let mut r = (0..e.n_features())
        .map(|j| WaterfallItem {
            feature: j,
            value: e.data()[[sample, j]],
            contribution: e.values()[[sample, j, output]],
        })
        .collect::<Vec<_>>();
    r.sort_by(|a, b| b.contribution.abs().total_cmp(&a.contribution.abs()));
    Ok(r)
}