def calc_probs(node: TrieNode) -> TrieNode:
new_root = TrieNode(node.id_, node_type=float)
new_root.words = {
word: node.apriori_prob(word)
for word in node.words.keys()
}
return new_root
def __init__(self, n: int):
self.n = n
self.root = TrieNode(IntVocabulary.extremal_element(), node_type=int)
self.lambdas = []
