File tree Expand file tree Collapse file tree 1 file changed +6
-3
lines changed Expand file tree Collapse file tree 1 file changed +6
-3
lines changed Original file line number Diff line number Diff line change @@ -191,7 +191,8 @@ cdef class DepthFirstTreeBuilder(TreeBuilder):
191
191
192
192
def __reduce__ (self ):
193
193
""" Reduce re-implementation, for pickling."""
194
- return (DepthFirstTreeBuilder, (self .splitter, self .min_samples_split,
194
+ return (DepthFirstTreeBuilder, (self .splitter,
195
+ self .min_samples_split,
195
196
self .min_samples_leaf,
196
197
self .min_weight_leaf,
197
198
self .max_depth,
@@ -686,9 +687,11 @@ cdef class BestFirstTreeBuilder(TreeBuilder):
686
687
687
688
def __reduce__ (self ):
688
689
""" Reduce re-implementation, for pickling."""
689
- return (BestFirstTreeBuilder, (self .splitter, self .min_samples_split,
690
+ return (BestFirstTreeBuilder, (self .splitter,
691
+ self .min_samples_split,
690
692
self .min_samples_leaf,
691
- self .min_weight_leaf, self .max_depth,
693
+ self .min_weight_leaf,
694
+ self .max_depth,
692
695
self .max_leaf_nodes,
693
696
self .min_impurity_decrease,
694
697
self .store_leaf_values,
You can’t perform that action at this time.
0 commit comments