quanteda (version 0.9.7-17)

predict.textmodel_NB_fitted: prediction method for Naive Bayes classifier objects

Description

implements class predictions using trained Naive Bayes examples

Usage

"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(LBGexample, c("A", "A", "B", "C", "C", NA)))
(nbpred <- predict(nbfit))