R/print.r

In ldelecourt/NBC_package: Original Implementation of Naive Bayes Classifier

#' Print NBAYES object attributes
#'
#' @description
#' print different information about an NBAYES object
#'
#' @usage
#' print.NBAYES(NBAYES)
#'
#' @param NBAYES an NBAYES object
#'
#' @return print all the different informations about the object NBAYES
#'
#' @examples
#' data(iris)
#' train <- iris
#' modele <- fit(Species ~ ., train)
#' print(modele)
#'
#' @export

#Surcharge de la methode print
print.NBAYES <- function(NBAYES){
  for (i in 1:length(NBAYES$table_proba_cond)){
    cat("Conditionnal Probabilities for the variable",names(NBAYES$table_proba_cond[i]),":","\n")
    print(NBAYES$table_proba_cond[[i]])
    cat("\n")
  }
  cat("Prior probability:","\n")
  print(NBAYES$prior)

  if (exists('NBAYES$cuts') && is.list('NBAYES$cuts')) {
    cat("Cuts:", "\n")
    print(NBAYES$cuts)
  }
}