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

use crate::{Explanation, Result, ShapError};
use ndarray::Array2;
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]
pub struct HeatmapData {
    pub values: Array2<f64>,
    pub feature_order: Vec<usize>,
}
pub fn data(e: &Explanation, output: usize) -> Result<HeatmapData> {
    if output >= e.n_outputs() {
        return Err(ShapError::InvalidOutputIndex {
            index: output,
            n_outputs: e.n_outputs(),
        });
    }
    let order = crate::plot::bar::data(e)
        .into_iter()
        .map(|x| x.0)
        .collect::<Vec<_>>();
    let values = Array2::from_shape_fn((e.n_samples(), e.n_features()), |(i, j)| {
        e.values()[[i, order[j], output]]
    });
    Ok(HeatmapData {
        values,
        feature_order: order,
    })
}