Revise set_h, h0, hn_params_recursion

yuta-nakahara · yuta-nakahara · commit 6f102fdfecd0 · 2022-12-16T22:31:09.000+09:00
diff --git a/bayesml/metatree/_metatree_x_discrete.py b/bayesml/metatree/_metatree_x_discrete.py
@@ -446,6 +446,7 @@ def _set_h_params_recursion(self,node:_Node,original_tree_node:_Node):
                             child_k_candidates,
                             sub_model=self.SubModel.GenModel(seed=self.rng,**self.sub_h_params),
                             )
+                    node.children[i].k_candidates = child_k_candidates
                     self._set_h_params_recursion(node.children[i],original_tree_node.children[i])
 
     def set_h_params(self,
@@ -914,6 +915,7 @@ def _set_h0_params_recursion(self,node:_Node,original_tree_node:_Node):
                             child_k_candidates,
                             sub_model=self.SubModel.LearnModel(**self.sub_h0_params),
                             )
+                    node.children[i].k_candidates = child_k_candidates
                     self._set_h0_params_recursion(node.children[i],original_tree_node.children[i])
 
     def _set_hn_g_recursion(self,node:_Node):
@@ -977,6 +979,7 @@ def _set_hn_params_recursion(self,node:_Node,original_tree_node:_Node):
                             child_k_candidates,
                             sub_model=self.SubModel.LearnModel(**self.sub_hn_params),
                             )
+                    node.children[i].k_candidates = child_k_candidates
                     self._set_hn_params_recursion(node.children[i],original_tree_node.children[i])
 
     def set_h0_params(self,

Original file line number	Diff line number	Diff line change
`@@ -446,6 +446,7 @@ def _set_h_params_recursion(self,node:_Node,original_tree_node:_Node):`
`446`	`446`	`child_k_candidates,`
`447`	`447`	`sub_model=self.SubModel.GenModel(seed=self.rng,**self.sub_h_params),`
`448`	`448`	`)`
	`449`	`+ node.children[i].k_candidates = child_k_candidates`
`449`	`450`	`self._set_h_params_recursion(node.children[i],original_tree_node.children[i])`
`450`	`451`
`451`	`452`	`def set_h_params(self,`
`@@ -914,6 +915,7 @@ def _set_h0_params_recursion(self,node:_Node,original_tree_node:_Node):`
`914`	`915`	`child_k_candidates,`
`915`	`916`	`sub_model=self.SubModel.LearnModel(**self.sub_h0_params),`
`916`	`917`	`)`
	`918`	`+ node.children[i].k_candidates = child_k_candidates`
`917`	`919`	`self._set_h0_params_recursion(node.children[i],original_tree_node.children[i])`
`918`	`920`
`919`	`921`	`def _set_hn_g_recursion(self,node:_Node):`
`@@ -977,6 +979,7 @@ def _set_hn_params_recursion(self,node:_Node,original_tree_node:_Node):`
`977`	`979`	`child_k_candidates,`
`978`	`980`	`sub_model=self.SubModel.LearnModel(**self.sub_hn_params),`
`979`	`981`	`)`
	`982`	`+ node.children[i].k_candidates = child_k_candidates`
`980`	`983`	`self._set_hn_params_recursion(node.children[i],original_tree_node.children[i])`
`981`	`984`
`982`	`985`	`def set_h0_params(self,`