coef: Extract Model Coefficients

In naivebayes: High Performance Implementation of the Naive Bayes Algorithm

Description

coef is a generic function which extracts model coefficients from specialized Naive Bayes objects.

Usage

## S3 method for class 'bernoulli_naive_bayes'
coef(object, ...)

## S3 method for class 'multinomial_naive_bayes'
coef(object, ...)

## S3 method for class 'poisson_naive_bayes'
coef(object, ...)

## S3 method for class 'gaussian_naive_bayes'
coef(object, ...)

Arguments

object	object of class inheriting from "*_naive_bayes".
...	not used.

Value

Coefficients extracted from the specialised Naive Bayes objects in form of a data frame.

Author(s)

Michal Majka, [e-mail]

See Also

bernoulli_naive_bayes, multinomial_naive_bayes, poisson_naive_bayes, gaussian_naive_bayes

Examples

data(iris)
y <- iris[[5]]
M <- as.matrix(iris[-5])

### Train the Gaussian Naive Bayes
gnb <- gaussian_naive_bayes(x = M, y = y)

### Extract coefficients
coef(gnb)

coef(gnb)[c(TRUE,FALSE)] # only means

coef(gnb)[c(FALSE,TRUE)] # only standard deviations

Example output

naivebayes 0.9.7 loaded
             setosa:mu setosa:sd versicolor:mu versicolor:sd virginica:mu
Sepal.Length     5.006 0.3524897         5.936     0.5161711        6.588
Sepal.Width      3.428 0.3790644         2.770     0.3137983        2.974
Petal.Length     1.462 0.1736640         4.260     0.4699110        5.552
Petal.Width      0.246 0.1053856         1.326     0.1977527        2.026
             virginica:sd
Sepal.Length    0.6358796
Sepal.Width     0.3224966
Petal.Length    0.5518947
Petal.Width     0.2746501
             setosa:mu versicolor:mu virginica:mu
Sepal.Length     5.006         5.936        6.588
Sepal.Width      3.428         2.770        2.974
Petal.Length     1.462         4.260        5.552
Petal.Width      0.246         1.326        2.026
             setosa:sd versicolor:sd virginica:sd
Sepal.Length 0.3524897     0.5161711    0.6358796
Sepal.Width  0.3790644     0.3137983    0.3224966
Petal.Length 0.1736640     0.4699110    0.5518947
Petal.Width  0.1053856     0.1977527    0.2746501

naivebayes documentation built on March 13, 2020, 1:31 a.m.