Skip to content

Commit e46581a

Browse files
authored
Merge pull request #49 from JuliaAI/change-default-ntrees
Change `n_trees` default to 100
2 parents ebd0c7c + 7d17b34 commit e46581a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/MLJDecisionTreeInterface.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,7 @@ MMI.@mlj_model mutable struct RandomForestClassifier <: MMI.Probabilistic
102102
min_samples_split::Int = 2::(_ ≥ 2)
103103
min_purity_increase::Float64 = 0.0::(_ ≥ 0)
104104
n_subfeatures::Int = (-)(1)::(_ ≥ -1)
105-
n_trees::Int = 10::(_ ≥ 2)
105+
n_trees::Int = 100::(_ ≥ 0)
106106
sampling_fraction::Float64 = 0.7::(0 < _ ≤ 1)
107107
feature_importance::Symbol = :impurity::(_ ∈ (:impurity, :split))
108108
rng::Union{AbstractRNG,Integer} = GLOBAL_RNG
@@ -304,7 +304,7 @@ MMI.@mlj_model mutable struct RandomForestRegressor <: MMI.Deterministic
304304
min_samples_split::Int = 2::(_ ≥ 2)
305305
min_purity_increase::Float64 = 0.0::(_ ≥ 0)
306306
n_subfeatures::Int = (-)(1)::(_ ≥ -1)
307-
n_trees::Int = 10::(_ ≥ 2)
307+
n_trees::Int = 100::(_ ≥ 0)
308308
sampling_fraction::Float64 = 0.7::(0 < _ ≤ 1)
309309
feature_importance::Symbol = :impurity::(_ ∈ (:impurity, :split))
310310
rng::Union{AbstractRNG,Integer} = GLOBAL_RNG

0 commit comments

Comments
 (0)