ldavis_json.R

ldavis_json <- function(m, dtm){

  dtm = dtm[slam::row_sums(dtm) > 0, ]
  
  # Find required quantities
  phi <- posterior(m)$terms %>% as.matrix
  theta <- posterior(m)$topics %>% as.matrix
  vocab <- colnames(phi)
  
  doc.length = slam::row_sums(dtm)
  term.freq = slam::col_sums(dtm)[match(vocab, colnames(dtm))]
  
  createJSON(phi = phi, theta = theta,
             vocab = vocab,
             doc.length = doc.length,
             term.frequency = term.freq)
  
}