R/mdlp_para.R

In ldelecourt/NBC_package: Original Implementation of Naive Bayes Classifier

#' Discretization
#'
#' @description
#' Discretization using the mdlp() function from the package discretization
#'
#' @usage
#' discret(colonne, Y)
#'
#' @param colonne the column of a dataframe that will be discretize, must be numeric
#' @param Y the column of a dataframe containing modalities to predict
#'
#' @return \strong{df_disc}, a dataframe with the variable colonne discretize
#' @return \strong{cuts}, a list giving the cuts made by the mdlp() function
#'
#'
#' @examples
#' data(iris)
#' column <- iris[1]
#' Y <- iris[5]
#' disc <- discret(column, Y)
#'
#'
#' @importFrom discretization mdlp
#'
#' @export

# Permet la discrétisation de la fonction mdlp() de la library(discretization)
# tout en retournant les cuts indispensable à la fonction predict().
# Cette fonction a été écrite pour rendre la parallélisation faisable dans la fonction fit().

discret <- function(colonne, Y) {
  df <- mdlp(cbind(colonne, Y))
  df_disc <- df$Disc.data[-2]
  cuts <- df$cutp
  return(list(df=df_disc, cuts=cuts))
}