From 6f62305a42f667bd522b942e8e9d89ef41e3a85e Mon Sep 17 00:00:00 2001 From: Robert Moss Date: Tue, 30 Jan 2024 10:23:52 -0800 Subject: [PATCH] Fixed `pruning_purity_threshold` default in docs --- src/scikitlearnAPI.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/scikitlearnAPI.jl b/src/scikitlearnAPI.jl index c1942cc2..e290465c 100644 --- a/src/scikitlearnAPI.jl +++ b/src/scikitlearnAPI.jl @@ -11,7 +11,7 @@ import ScikitLearnBase: # Classifier """ - DecisionTreeClassifier(; pruning_purity_threshold=0.0, + DecisionTreeClassifier(; pruning_purity_threshold=1.0, max_depth::Int=-1, min_samples_leaf::Int=1, min_samples_split::Int=2, @@ -132,7 +132,7 @@ end # Regression """ - DecisionTreeRegressor(; pruning_purity_threshold=0.0, + DecisionTreeRegressor(; pruning_purity_threshold=1.0, max_depth::Int-1, min_samples_leaf::Int=5, min_samples_split::Int=2,