Random forest model with randomForest
Usage
pipe_randomForest(
df,
predInput = NULL,
responseVar = 1,
caseClass = NULL,
idVars = character(),
weight = "class",
crossValStrategy = c("Kfold", "bootstrap"),
k = 5,
replicates = 10,
crossValRatio = c(train = 0.6, test = 0.2, validate = 0.2),
ntree = 500,
importance = TRUE,
shap = TRUE,
aggregate_shap = TRUE,
repVi = 5,
summarizePred = TRUE,
scaleDataset = FALSE,
RFmodel = FALSE,
DALEXexplainer = FALSE,
variableResponse = FALSE,
save_validateset = FALSE,
filenameRasterPred = NULL,
tempdirRaster = NULL,
nCoresRaster = parallel::detectCores()%/%2,
verbose = 0,
...
)
Arguments
- df
a
data.frame
with the data.- predInput
a
data.frame
or aRaster
with the input variables for the model as columns or layers. The columns or layer names must match the names ofdf
columns.- responseVar
response variable as column name or index on
df
.- caseClass
class of the samples used to weight cases. Column names or indexes on
df
, or a vector with the class for each rows indf
.- idVars
id column names or indexes on
df
. This columns will not be used for training.- weight
Optional array of the same length as
nrow(df)
, containing weights to apply to the model's loss for each sample.- crossValStrategy
Kfold
orbootstrap
.- k
number of data partitions when
crossValStrategy="Kfold"
.- replicates
number of replicates for
crossValStrategy="bootstrap"
andcrossValStrategy="Kfold"
(replicates * k-1
, 1 fold for validation).- crossValRatio
Proportion of the dataset used to train, test and validate the model when
crossValStrategy="bootstrap"
. Default toc(train=0.6, test=0.2, validate=0.2)
. If there is only one value, will be taken as a train proportion and the test set will be used for validation.- ntree
Number of trees to grow.
- importance
parameter for
randomForest::randomForest()
indicating if importance of predictors should be assessed.- shap
if
TRUE
, return the SHAP values asshapviz::shapviz()
objects.- aggregate_shap
if
TRUE
, andshap
is alsoTRUE
, aggregate SHAP from all replicates.- repVi
replicates of the permutations to calculate the importance of the variables. 0 to avoid calculating variable importance.
- summarizePred
if
TRUE
, return the mean, sd and se of the predictors. ifFALSE
, return the predictions for each replicate.- scaleDataset
if
TRUE
, scale the whole dataset only once instead of the train set at each replicate. Optimize processing time for predictions with large rasters.- RFmodel
if TRUE, return the model with the result.
- DALEXexplainer
if
TRUE
, return a explainer for the models fromDALEX::explain()
function. It doesn't work with multisession future plans.- variableResponse
if
TRUE
, return aggregated_profiles_explainer object fromingredients::partial_dependency()
and the coefficients of the adjusted linear model.- save_validateset
save the validateset (independent data not used for training).
- filenameRasterPred
if no missing, save the predictions in a RasterBrick to this file.
- tempdirRaster
path to a directory to save temporal raster files.
- nCoresRaster
number of cores used for parallelized raster cores. Use half of the available cores by default.
- verbose
If > 0, print state and passed to randomForest functions
- ...
extra parameters for
randomForest::randomForest()
,future.apply::future_replicate()
andingredients::feature_importance()
.