朴素贝叶斯算法在R语言中的应用,对应klaR包中的NaiveBayes()方法。问题描述:主要通过朴素贝叶斯算法对于测试数据集中的nmkat属性值进行预测,我们使用的数据是KKNN包中的自带数据miete数据。主要对于该数据进行划分:1/3为测试数据,2/3为训练数据,通过对于数据的观察,预测测试数据中的nmkat对应的值。
加载包
#朴素贝叶斯算法
library(klaR)
nb1=NaiveBayes(nmkat~.,data_train)
划分数据集
data_train=getdata(miete[,c(-1,-3,-12)],sub_train$ID_unit)
data_test=getdata(miete[,c(-1,-3,-12)],-sub_train$ID_unit)
test_nb=NaiveBayes(data_train[,-12],data_train[,12])
预测