Skip to content

Instantly share code, notes, and snippets.

@toyeiei
Last active November 13, 2018 04:51
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save toyeiei/ed7525fbaf9e88760387033d01f70971 to your computer and use it in GitHub Desktop.
Save toyeiei/ed7525fbaf9e88760387033d01f70971 to your computer and use it in GitHub Desktop.
random forest in R
# install.packages("randomForest")
library(randomForest)
# build random forest model
set.seed(123)
rf_model <- randomForest(Class ~ ., data = train_df)
# print model
print(rf_model)
# predict test data and compute accuracy
p <- predict(rf_model, newdata = test_df)
table(test_df$Class, p)
# p
# benign malignant
# benign 127 4
# malignant 1 73
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment