predict.textmodel_nb: Prediction from a fitted textmodel_nb object

Description Usage Arguments Value See Also Examples

View source: R/textmodel_nb.R

Description

predict.textmodel_nb() implements class predictions from a fitted Naive Bayes model. using trained Naive Bayes examples

Usage

1
2
3
## S3 method for class 'textmodel_nb'
predict(object, newdata = NULL,
  type = c("class", "probability", "logposterior"), force = FALSE, ...)

Arguments

object

a fitted Naive Bayes textmodel

newdata

dfm on which prediction should be made

type

the type of predicted values to be returned; see Value

force

make newdata's feature set conformant to the model terms

...

not used

Value

predict.textmodel_nb returns either a vector of class predictions for each row of newdata (when type = "class"), or a document-by-class matrix of class probabilities (when type = "probability") or log posterior likelihoods (when type = "logposterior").

See Also

textmodel_nb

Examples

1
2
3
4
# application to LBG (2003) example data
(tmod <- textmodel_nb(data_dfm_lbgexample, y = c("A", "A", "B", "C", "C", NA)))
predict(tmod)
predict(tmod, type = "logposterior")

quanteda/quanteda documentation built on Feb. 16, 2019, 5:45 a.m.