A wrapper for the CTM function based on Blei's original code that returns a nicely-formatted topic model.

FitCtmModel(dtm, k, calc_coherence = TRUE, calc_r2 = FALSE,
  return_all = TRUE, ...)

Arguments

dtm

A document term matrix of class dgCMatrix

k

Number of topics

calc_coherence

Do you want to calculate probabilistic coherence of topics after the model is trained? Defaults to TRUE.

calc_r2

Do you want to calculate R-squared after the model is trained? Defaults to FALSE.

return_all

Logical. Do you want the raw results of the underlying function returned along with the formatted results? Defaults to TRUE.

...

Other arguments to pass to CTM or TmParallelApply.

Value

Returns a list with a minimum of two objects, phi and theta. The rows of phi index topics and the columns index tokens. The rows of theta index documents and the columns index topics.

Examples

# Load a pre-formatted dtm data(nih_sample_dtm) # Fit a CTM model on a sample of documents model <- FitCtmModel(dtm = nih_sample_dtm[ sample(1:nrow(nih_sample_dtm) , 10) , ], k = 3, return_all = FALSE)