data(iris)
slevels <- levels(iris$Species)
model <- cla_rf("Species", slevels, ntree=5)
# preparing dataset for random sampling
sr <- sample_random()
sr <- train_test(sr, iris)
train <- sr$train
test <- sr$test
model <- fit(model, train)
prediction <- predict(model, test)
predictand <- adjust_class_label(test[,"Species"])
test_eval <- evaluate(model, predictand, prediction)
test_eval$metrics
Run the code above in your browser using DataLab