File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments