reg_rf | R Documentation |
Creates a regression object that uses the Random Forest method. It wraps the randomForest library.
reg_rf(attribute, nodesize = 1, ntree = 10, mtry = NULL)
attribute |
attribute target to model building |
nodesize |
node size |
ntree |
number of trees |
mtry |
number of attributes to build tree |
returns an object of class reg_rf
obj
data(Boston)
model <- reg_rf("medv", ntree=10)
# preparing dataset for random sampling
sr <- sample_random()
sr <- train_test(sr, Boston)
train <- sr$train
test <- sr$test
model <- fit(model, train)
test_prediction <- predict(model, test)
test_predictand <- test[,"medv"]
test_eval <- evaluate(model, test_predictand, test_prediction)
test_eval$metrics
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.