shapley
is a fast Shapley value calculator written in rust.
- Python:
pip install shapleyrs - Rust:
cargo add shapley
from shapleyrs import Shapley
shapley = Shapley(players=[1, 2], coalition_worth={
(): 0,
(1,): 10,
(2,): 20,
(1, 2): 30,
})
print(shapley.shapley_value(1)) # 10
print(shapley.shapley_value(2)) # 20use std::collections::HashMap;
use shapley::{Coalition, Shapley};
fn main() {
let players = vec![1, 2];
let coalition_worth = HashMap::from([
(Coalition::new(vec![]), 0.0),
(Coalition::new(vec![1]), 10.0),
(Coalition::new(vec![2]), 20.0),
(Coalition::new(vec![1, 2]), 30.0),
]);
let shapley = Shapley::new(players, coalition_worth);
println!("player1: {}", shapley.shapley_value(1).unwrap()); // 10
println!("player2: {}", shapley.shapley_value(2).unwrap()); // 20
}