fanova

一个 Rust fANOVA (函数分析方差) 实现。

fANOVA 提供了一种计算特征重要性的方法。

示例

use fanova::{FanovaOptions, RandomForestOptions};
use rand::{Rng, SeedableRng};

let mut feature1 = Vec::new();
let mut feature2 = Vec::new();
let mut feature3 = Vec::new();
let mut target = Vec::new();

let mut rng = rand::rngs::StdRng::seed_from_u64(0);
for _ in 0..100 {
    let f1 = rng.gen();
    let f2 = rng.gen();
    let f3 = rng.gen();
    let t = f1 + f2 * 2.0 + f3 * 3.0;

    feature1.push(f1);
    feature2.push(f2);
    feature3.push(f3);
    target.push(t);
}

let mut fanova = FanovaOptions::new()
    .random_forest(RandomForestOptions::new().seed(0))
    .fit(vec![&feature1, &feature2, &feature3], &target).unwrap();
let importances = (0..3)
    .map(|i| fanova.quantify_importance(&[i]).mean)
    .collect::<Vec<_>>();

assert_eq!(
    importances,
    vec![0.02744461966313835, 0.22991883769286145, 0.6288784011550144]
);

参考文献

• 评估超参数重要性的有效方法