# training data set
data(x_train)
# training data label
data(y_train)
# test data set
data(x_test)
# test data label
data(y_test)
library(e1071)
#svm
svm_model = svm(x_train,y_train)
y_pred_svm = predict(svm_model,x_test)
y_pred_svm = round(y_pred_svm)
accuracy_svm = sum(y_pred_svm==y_test)/dim(y_test)[1]
#naive bayes
bayes_model = naiveBayes(x_train,y_train)
y_pred_bayes = predict(bayes_model,x_test)
y_pred_bayes = round(y_pred_bayes)
accuracy_bayes = sum(y_pred_bayes==y_test)/dim(y_test)[1]
标签是从 1 到 5 的数字。
此代码适用于 SVM,但不适用于朴素贝叶斯分类器,更具体地说,y_pred_svm 是数字向量,但 y_pred_bayes 不是。
我在包中找到了很多关于 naiveBayes 函数的示例,但数据和标签不在单独的数组中。谁能帮我吗?谢谢。