我正在使用预测和NaiveBayes运行一个天真的贝叶斯,其中包含一些预测变量和二元结果(已失效或有效)。 NB运行正常,但是当涉及到混淆矩阵时,我得到错误:
Error in predictions$class : $ operator is invalid for atomic vectors
你知道怎么纠正这个吗?数据作为csv读入,这里是代码:
trainIndex <- createDataPartition(thl_new$Lapsed, p=0.65, list=FALSE)
data_train <- thl_new[ trainIndex,]
data_test <- thl_new[-trainIndex,]
model <- naiveBayes(Lapsed~., data=data_train, laplace = 1)
predictions <- predict(model, data_test[,1:5]) # test accuracy on 'data_test'
predictions
confusionMatrix(predictions$class, data_test$Lapsed) # show new data estimates