Use of randomforest() for classification in R?

marc picture marc · Oct 10, 2013 · Viewed 13.2k times · Source

I originally had a data frame composed of 12 columns in N rows. The last column is my class (0 or 1). I had to convert my entire data frame to numeric with

training <- sapply(training.temp,as.numeric)

But then I thought I needed the class column to be a factor column to use the randomforest() tool as a classifier, so I did

training[,"Class"] <- factor(training[,ncol(training)])

I proceed to creating the tree with

training_rf <- randomForest(Class ~., data = trainData, importance = TRUE, do.trace = 100)

But I'm getting two errors:

1: In Ops.factor(training[, "Status"], factor(training[, ncol(training)])) : 
<= this is not relevant for factors (roughly translated)
2: In randomForest.default(m, y, ...) :
The response has five or fewer unique values.  Are you sure you want to do regression?

I would appreciate it if someone could point out the formatting mistake I'm making.

Thanks!

Answer

marc picture marc · Oct 11, 2013

So the issue is actually quite simple. It turns out my training data was an atomic vector. So it first had to be converted as a data frame. So I needed to add the following line:

training <- as.data.frame(training)

Problem solved!