0
# 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 函数的示例,但数据和标签不在单独的数组中。谁能帮我吗?谢谢。

4

0 回答 0