Fix grf tree

lixfz · lixfz · commit d6fefbce286c · 2023-01-03T19:02:19.000+08:00
diff --git a/ylearn/estimator_model/_generalized_forest/tree/_grf_tree.py b/ylearn/estimator_model/_generalized_forest/tree/_grf_tree.py
@@ -331,7 +331,7 @@ def _fit_with_array(self, x, y, w, v, sample_weight=None):
         self.max_features_ = max_features
 
         max_leaf_nodes = (
-            np.int(1e3) if self.max_leaf_nodes is None else self.max_leaf_nodes
+            1000 if self.max_leaf_nodes is None else self.max_leaf_nodes
         )
 
         if len(y) != n_samples or len(x) != n_samples:

Original file line number	Diff line number	Diff line change
`@@ -331,7 +331,7 @@ def _fit_with_array(self, x, y, w, v, sample_weight=None):`
`331`	`331`	`self.max_features_ = max_features`
`332`	`332`
`333`	`333`	`max_leaf_nodes = (`
`334`		`- np.int(1e3) if self.max_leaf_nodes is None else self.max_leaf_nodes`
	`334`	`+ 1000 if self.max_leaf_nodes is None else self.max_leaf_nodes`
`335`	`335`	`)`
`336`	`336`
`337`	`337`	`if len(y) != n_samples or len(x) != n_samples:`