朴素贝叶斯使用NaiveBayes和预测 - 原子矢量?

时间:2015-11-18 10:28:43

标签: r r-caret

我正在使用预测和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

0 个答案:

没有答案