quanteda (version 0.9.9-50)

predict.textmodel_NB_fitted: prediction method for Naive Bayes classifier objects

Description

implements class predictions using trained Naive Bayes examples

Usage

# S3 method for textmodel_NB_fitted
predict(object, newdata = NULL, ...)

Arguments

object

a fitted Naive Bayes textmodel

newdata

dfm on which prediction should be made

...

not used

Value

A list of two data frames, named docs and words corresponding to word- and document-level predicted quantities

docs

data frame with document-level predictive quantities: nb.predicted, ws.predicted, bs.predicted, PcGw, wordscore.doc, bayesscore.doc, posterior.diff, posterior.logdiff. Note that the diff quantities are currently implemented only for two-class solutions.

words

data-frame with word-level predictive quantities: wordscore.word, bayesscore.word

Examples

(nbfit <- textmodel_NB(data_dfm_LBGexample, c("A", "A", "B", "C", "C", NA)))
(nbpred <- predict(nbfit))