Skip to contents

Make prediction from a fine-tuned grafzahl object.

Usage

# S3 method for grafzahl
predict(object, newdata, cuda = detect_cuda(), return_raw = FALSE, ...)

Arguments

object

an S3 object trained with grafzahl()

newdata

a corpus or a character vector of texts on which prediction should be made.

cuda

logical, whether to use CUDA, default to detect_cuda().

return_raw

logical, if TRUE, return a matrix of logits; a vector of class prediction otherwise

...

not used

Value

a vector of class prediction or a matrix of logits