Nothing
#' @title Naive Bayes Classification Learner
#'
#' @name mlr_learners_classif.naive_bayes
#'
#' @description
#' Naive Bayes classification.
#' Calls [e1071::naiveBayes()] from package \CRANpkg{e1071}.
#'
#' @templateVar id classif.naive_bayes
#' @template learner
#'
#' @export
#' @template seealso_learner
#' @template example
LearnerClassifNaiveBayes = R6Class("LearnerClassifNaiveBayes",
inherit = LearnerClassif,
public = list(
#' @description
#' Creates a new instance of this [R6][R6::R6Class] class.
initialize = function() {
ps = ps(
eps = p_dbl(default = 0, tags = "predict"),
laplace = p_dbl(0, default = 0, tags = "train"),
threshold = p_dbl(default = 0.001, tags = "predict")
)
super$initialize(
id = "classif.naive_bayes",
param_set = ps,
predict_types = c("response", "prob"),
properties = c("twoclass", "multiclass"),
feature_types = c("logical", "integer", "numeric", "factor"),
packages = c("mlr3learners", "e1071"),
label = "Naive Bayes",
man = "mlr3learners::mlr_learners_classif.naive_bayes"
)
}
),
private = list(
.train = function(task) {
y = task$truth()
x = task$data(cols = task$feature_names)
invoke(e1071::naiveBayes,
x = x, y = y,
.args = self$param_set$get_values(tags = "train"))
},
.predict = function(task) {
pv = self$param_set$get_values(tags = "predict")
newdata = ordered_features(task, self)
if (self$predict_type == "response") {
response = invoke(predict, self$model,
newdata = newdata,
type = "class", .args = pv)
list(response = response)
} else {
prob = invoke(predict, self$model, newdata = newdata,
type = "raw", .args = pv)
list(prob = prob)
}
}
)
)
#' @include aaa.R
learners[["classif.naive_bayes"]] = LearnerClassifNaiveBayes
Any scripts or data that you put into this service are public.
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.