Skip to content

Commit 6f102fd

Browse files
committed
Revise set_h, h0, hn_params_recursion
1 parent 09ebf4f commit 6f102fd

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

bayesml/metatree/_metatree_x_discrete.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -446,6 +446,7 @@ def _set_h_params_recursion(self,node:_Node,original_tree_node:_Node):
446446
child_k_candidates,
447447
sub_model=self.SubModel.GenModel(seed=self.rng,**self.sub_h_params),
448448
)
449+
node.children[i].k_candidates = child_k_candidates
449450
self._set_h_params_recursion(node.children[i],original_tree_node.children[i])
450451

451452
def set_h_params(self,
@@ -914,6 +915,7 @@ def _set_h0_params_recursion(self,node:_Node,original_tree_node:_Node):
914915
child_k_candidates,
915916
sub_model=self.SubModel.LearnModel(**self.sub_h0_params),
916917
)
918+
node.children[i].k_candidates = child_k_candidates
917919
self._set_h0_params_recursion(node.children[i],original_tree_node.children[i])
918920

919921
def _set_hn_g_recursion(self,node:_Node):
@@ -977,6 +979,7 @@ def _set_hn_params_recursion(self,node:_Node,original_tree_node:_Node):
977979
child_k_candidates,
978980
sub_model=self.SubModel.LearnModel(**self.sub_hn_params),
979981
)
982+
node.children[i].k_candidates = child_k_candidates
980983
self._set_hn_params_recursion(node.children[i],original_tree_node.children[i])
981984

982985
def set_h0_params(self,

0 commit comments

Comments
 (0)