rf | R Documentation |
Implemented regression methods
rf(y, x, ...)
survforest(y, x, ...)
qrf(y, x, ...)
lrm(y, x, ...)
glrm(y, x, ...)
lasso(y, x, ...)
ridge(y, x, ...)
postlasso(y, x, ...)
cox(y, x, ...)
tuned_rf(
y,
x,
max.depths = 1:5,
mtrys = list(1, function(p) ceiling(sqrt(p)), identity),
verbose = FALSE,
...
)
xgb(y, x, nrounds = 2, verbose = 0, ...)
tuned_xgb(
y,
x,
etas = c(0.1, 0.5, 1),
max_depths = 1:5,
nfold = 5,
nrounds = c(2, 10, 50),
verbose = 0,
metrics = list("rmse"),
...
)
y |
Vector (or matrix) of response values. |
x |
Design matrix of predictors. |
... |
Additional arguments passed to the underlying regression method.
In case of |
max.depths |
Values for |
mtrys |
for |
verbose |
See |
nrounds |
See |
etas |
Values for |
max_depths |
Values for |
nfold |
Number of folds for |
metrics |
See |
The implemented choices are "rf"
for random forests as implemented in
ranger, "lasso"
for cross-validated Lasso regression (using the
one-standard error rule), "ridge"
for cross-validated ridge regression (using the one-standard error rule),
"cox"
for the Cox proportional
hazards model as implemented in survival, "qrf"
or "survforest"
for quantile and survival random forests, respectively. The option
"postlasso"
option refers to a cross-validated LASSO (using the
one-standard error rule) and subsequent OLS regression. The "lrm"
option implements a standard linear regression model. The "xgb"
and
"tuned_xgb"
options require the xgboost
package.
The "tuned_rf"
regression method tunes the mtry
and
max.depth
parameters in ranger
out-of-bag.
The "tuned_xgb"
regression method uses k-fold cross-validation to
tune the nrounds
, mtry
and max_depth
parameters in
xgb.cv
.
New regression methods can be implemented and supplied as well and need the
following structure. The regression method "custom_reg"
needs to take
arguments y, x, ...
, fit the model using y
and x
as
matrices and return an object of a user-specified class, for instance,
'custom
'. For the GCM test, implementing a residuals.custom
method is sufficient, which should take arguments
object, response = NULL, data = NULL, ...
. For the PCM test, a
predict.custom
method is necessary for out-of-sample prediction
and computation of residuals.
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.