extract_params.naiveBayes {aurelius}R Documentation

extract_params.naiveBayes

Description

Extracts a parameters from an ensemble made by the naiveBayes function

Usage

## S3 method for class 'naiveBayes'
extract_params(object, threshold = 0.001, eps = 0, ...)

Arguments

object

an object of class "naiveBayes"

threshold

a value replacing cells with probabilities within eps range.

eps

a numeric for specifying an epsilon-range to apply laplace smoothing (to replace zero or close-zero probabilities by theshold.)

...

further arguments passed to or from other methods

Value

a list that is extracted from the naiveBayes object

Examples

model <- e1071::naiveBayes(Species ~ ., data=iris) 
model_params <- extract_params(model)

[Package aurelius version 0.8.4 Index]