朴素貝葉斯分類器期間出錯

Question

我有一個5000點的數據集和12個屬性（其中不是類變量）..我將數據分為訓練（3000點）和測試（2000點），並對訓練數據和wnat進行分類以檢查錯誤率使用准確性指標，但很遺憾，正在拋出錯誤，請您幫我一下。

b=as.factor(test_data$Personal.Loan)
model_naivebayes = naiveBayes(Personal.Loan ~.,data=train_data);
naive_predict = predict(model_naivebayes, test_data);
table(naive_predict,b)

錯誤：表（naive_predict，b）中的錯誤：所有參數的長度必須相同

當我檢查naive_predict中的內容時，它說因子W /'0'

問候，斯里。

Answer 1

我有類似的問題，並以此方式解決了。 我將用iris數據顯示它：

此代碼將給出錯誤消息：

 iris[ , 5] <- as.character(iris[ , 5] )
 classifier<-naiveBayes(iris[,1:4], iris[,5]) 
 table(predict(classifier, iris[,-5]), iris[,5])

如果使用factor，則不會：

 iris[ , 5] <- as.character(iris[ , 5] )
 classifier<-naiveBayes(iris[,1:4], factor(iris[,5]) )
 table(predict(classifier, iris[,-5]), factor(iris[,5]))

Answer 2

看起來錯誤在第三行。 預測時需要排除類變量。

naive_predict = predict(model_naivebayes, test_data[,-which(names(predictors) %in% c("Personal.Loans"))];