Learn R Programming

BERTopic (version 0.1.0)

predict.bertopic_r: Predict method for BERTopic models

Description

Predict method for BERTopic models

Usage

# S3 method for bertopic_r
predict(
  object,
  newdata,
  type = c("both", "class", "prob"),
  embeddings = NULL,
  ...
)

Value

Depending on type, an integer vector, a matrix/data frame, or a list.

Arguments

object

A "bertopic_r" model.

newdata

Character vector of new documents.

type

One of "class", "prob", or "both".

embeddings

Optional numeric matrix of embeddings.

...

Reserved for future arguments.