Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- library(e1071)
- library(caret)
- library(RWeka)
- library(mi)
- DB <- read.csv("/home/onu/Desktop/ML/Imtiaz/data.csv")
- DB$class <- as.character(DB$class)
- DB$class <- as.factor(DB$class)
- DB$ind <- sample(2,nrow(DB),replace=TRUE,prob=c(0.7,0.3))
- trainData <- DB[(DB$ind ==1),]
- testData <-DB[(DB$ind ==2),]
- NB <- naiveBayes(class~age+gender+tB+dB+alkphos+sgpt+sgot+tP+al.B+aG, data=trainData)
- predNB <- predict(NB,testData,type=c("class"))
- nb_mat <- table(testData$class,predNB)
- nb_mat
- predictionNB<-predict(NB,testData)
- cfMatrixNB<-confusionMatrix(testData$class,predictionNB)
- cfMatrixNB
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement