matthieu/pve-network
1
0
Fork 0
forked from Mirror/pve-network
Code Pull requests Activity

make weighted probabilities depending on node's children children count

Browse source
This commit is contained in:
Matthieu Pignolet 2024-06-06 14:15:04 +04:00
parent e3f5437ed8
commit 16e9f11a3d
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
autofeur_db/src/trie.rs
View file

@ -109,11 +109,11 @@ impl<'a> Trie<'a> {
while current_node.child_nodes.len() != 0 { while current_node.child_nodes.len() != 0 {
// We need to choose a random child based on weights // We need to choose a random child based on weights
let weighted = WeightedIndex::new(current_node.child_nodes.iter().map(|(_, node)| { let weighted = WeightedIndex::new(current_node.child_nodes.iter().map(|(_, node)| {
node.child_count / node node.child_count / (node
.child_nodes .child_nodes
.iter() .iter()
.map(|(_, b)| b.child_count) .map(|(_, b)| b.child_count)
.sum::<u64>() .sum::<u64>() + 1)
})) }))
.expect("distribution creation should be valid"); .expect("distribution creation should be valid");