Advertisement
Guest User

naiveBayes

a guest
Feb 20th, 2019
80
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.58 KB | None | 0 0
  1. library(e1071)
  2. library(caret)
  3. library(RWeka)
  4. library(mi)
  5. DB <- read.csv("/home/onu/Desktop/ML/Imtiaz/data.csv")
  6. DB$class <- as.character(DB$class)
  7. DB$class <- as.factor(DB$class)
  8. DB$ind <- sample(2,nrow(DB),replace=TRUE,prob=c(0.7,0.3))
  9. trainData <- DB[(DB$ind ==1),]
  10. testData <-DB[(DB$ind ==2),]
  11. NB <- naiveBayes(class~age+gender+tB+dB+alkphos+sgpt+sgot+tP+al.B+aG, data=trainData)
  12. predNB <- predict(NB,testData,type=c("class"))
  13. nb_mat <- table(testData$class,predNB)
  14. nb_mat
  15. predictionNB<-predict(NB,testData)
  16. cfMatrixNB<-confusionMatrix(testData$class,predictionNB)
  17. cfMatrixNB
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement