下面是朴素bayes的一些代码(某一课程
老师写的)
library(bnlearn)
setwd("D:\\R_edu\\data")
orgData<-read.csv("date_data2.csv")
summary(orgData)
rm(orgData1)
orgData1<-orgData[,c("income_rank","attractive_rank","assets_rank","Dated")]
orgData1$income_rank=as.factor(orgData$income_rank)
orgData1$attractive_rank=as.factor(orgData$attractive_rank)
orgData1$assets_rank=as.factor(orgData$assets_rank)
orgData1$Dated=as.factor(orgData$Dated)
bn = naive.bayes(orgData1, "Dated")
pred = predict(bn, orgData1,prob=T)
orgData1$pred=attr(pred,"prob")[,2]
关于对预测概率的提取,我觉得写法不对,应该是以下写法(orgData1$pred=attr(pred,"prob")[2,]),请大家帮忙指出到底该怎么写。