Легенда для Random Forest земли в R

Question

Я создал модель случайного леса предсказания в R с использованием RandomForest функции:

model = randomForest(classification ~., data=train, ntree=100, proximity=T) 

Далее я построена модель для того, чтобы увидеть общую погрешность модели:

plot(model, log="y") 

Это дает мне следующий сюжет:

Мой вопрос, как я могу поставить легенду об этом, чтобы я мог увидеть, с olor соответствует каждому значению в коэффициенте, используемом для классификации? Коэффициентная переменная равна data$classification. Я не могу понять вызов legend(), чтобы сделать это.

Answer 1

Сюжет участка участка S3 использует matplot для построения случайной модели леса. Вы должны добавить легенду вручную. Это должно быть хорошим началом:

library(randomForest) 
model = randomForest(Species ~., data=iris, ntree=100, proximity=T) 
layout(matrix(c(1,2),nrow=1), 
     width=c(4,1)) 
par(mar=c(5,4,4,0)) #No margin on the right side 
plot(model, log="y") 
par(mar=c(5,0,4,2)) #No margin on the left side 
plot(c(0,1),type="n", axes=F, xlab="", ylab="") 
legend("top", colnames(model$err.rate),col=1:4,cex=0.8,fill=1:4) 

Answer 2

вы можете использовать это,

model$finalModel.legend <- if (is.null(model$finalModel$test$err.rate)) {colnames(model$finalModel$err.rate)} else {colnames(model$finalModel$test$err.rate)} 
legend("top", cex =0.7, legend=model$finalModel.legend, lty=c(1,2,3), col=c(1,2,3), horiz=T)