Skip to content
This repository has been archived by the owner on Jan 6, 2022. It is now read-only.

Commit

Permalink
ntree is not a tuning parameter
Browse files Browse the repository at this point in the history
  • Loading branch information
jakob-r committed May 29, 2017
1 parent 948f501 commit cf83f8a
Showing 1 changed file with 0 additions and 4 deletions.
4 changes: 0 additions & 4 deletions R/getDefaultParSetValues.R
Original file line number Diff line number Diff line change
Expand Up @@ -88,26 +88,22 @@ getDefaultParSetValues = function() {
## not compared to caret
# random forest (only mtry in caret)
classif.randomForest = makeParamSet(
makeNumericParam("ntree", lower = log2(10/10), upper = log2(1000/10), trafo = function(x) round(2^x * 10), default = log2(500/10)),
makeIntegerParam("nodesize", lower = 1, upper = 10, default = 1),
makeIntegerParam("mtry", lower = 1L, upper = expression(p), default = expression(floor(sqrt(p)))),
keys = "p"
),
regr.randomForest = makeParamSet(
makeNumericParam("ntree", lower = log2(10/10), upper = log2(1000/10), trafo = function(x) round(2^x * 10), default = log2(500/10)),
makeIntegerParam("nodesize", lower = 1, upper = 10, default = 1),
makeIntegerParam(id = "mtry", lower = 1L, upper = expression(p), default = expression(max(floor(p/3), 1))),
keys = "p"
),
classif.ranger = makeParamSet(
makeIntegerParam("mtry", lower = 1L, upper = expression(p), default = expression(floor(sqrt(p)))),
makeNumericParam("num.trees", lower = 0, upper = 7, trafo = function(x) round(2^x * 10), default = log2(500/10)),
makeIntegerParam("min.node.size", lower = 1, upper = 10, default = 1),
keys = "p"
),
regr.ranger = makeParamSet(
makeIntegerParam("mtry", lower = 1L, upper = expression(p), default = expression(max(floor(p/3), 1))),
makeNumericParam("num.trees", lower = 0, upper = 7, trafo = function(x) round(2^x * 10), default = log2(500/10)),
makeIntegerParam("min.node.size", lower = 1, upper = 10, default = 5),
keys = "p"
),
Expand Down

0 comments on commit cf83f8a

Please sign in to comment.