앙상블 모형 제작 in r

9566·2021년 6월 9일
1

데이터 분석

목록 보기
4/10

3개 모델로 앙상블 모형 제작

model1 <- rf.model
model2 <- svm
model3 <- knn
pred1 <- predict(model1, data)
pred2 <- predict(model2, data)
pred3 <- predict(model3, data)
# 각각의 예측 결과를 취합
test <- data.frame(Y = data$Y, pred1, pred2, pred3)
head(test)
# 3개 분류기의 결과를 취합하여 최종 결과를 voting
funcResultValue <- function(x) {
    result <- NULL
    for (i in 1:nrow(x)) {
        xtab <- table(t(x[i, ]))
        rvalue <- names(sort(xtab, decreasing = T)[1])
        result <- c(result, rvalue)
    }
    return(result)
}
test$result <- funcResultValue(test[ ,2:4])
confusionMatrix(test$result, test$Species)
test$result # 앙상블 모형 예측값
test$result <- as.dataframe(test$result)
write.csv(test$result, "result.csv")

profile
안녕하세요 안녕안녕하세요 안녕하세요오오오~~

0개의 댓글