From e3f5437ed889dd7b736a891144b8123d2b92a81b Mon Sep 17 00:00:00 2001 From: Matthieu Pignolet Date: Thu, 6 Jun 2024 14:13:41 +0400 Subject: [PATCH] make weighted probabilities depending on node's children children count --- autofeur_db/src/trie.rs | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/autofeur_db/src/trie.rs b/autofeur_db/src/trie.rs index bbc2a93..6d0867a 100644 --- a/autofeur_db/src/trie.rs +++ b/autofeur_db/src/trie.rs @@ -108,12 +108,13 @@ impl<'a> Trie<'a> { let mut length = 0; while current_node.child_nodes.len() != 0 { // We need to choose a random child based on weights - let weighted = WeightedIndex::new( - current_node + let weighted = WeightedIndex::new(current_node.child_nodes.iter().map(|(_, node)| { + node.child_count / node .child_nodes .iter() - .map(|(_, node)| node.child_count / (length + 1)), - ) + .map(|(_, b)| b.child_count) + .sum::() + })) .expect("distribution creation should be valid"); let (key, node) = current_node -- 2.39.5