R/matrix_gen.R

In mfinst/TM-CoCit-Support-FM: What the package does (short line)

matrix_gen <-
function(corpus = NULL, structKey = 'm', dtmTokenizer = NULL) {
  
  # Die DocumentTermMatrix wird zuerst erstellt
  if (!is.null(dtmTokenizer)) {
    dtm = DocumentTermMatrix(corpus, control = list(tokenize = phraseTokenizer))
  } else {
    dtm = DocumentTermMatrix(corpus)
  }
  if(structKey == 'dtm'){
    return(dtm)
  }
  
  # Matrix Berechnung
  m = as.matrix(dtm)
  if(structKey == 'm') {
    return(m)
  }
  v = sort(colSums(m), decreasing=TRUE)
  if(structKey == 'v') {
    return(v)
  }
  
  # TF Matrix Berechnung
  d = data.frame(word = names(v), freq = v)
  if(structKey == 'd' || structKey == 'tf') {
    return(d)
  }
  
  # TF-IDF Berechnung
  dtm.tfidf = weightTfIdf(dtm)
  dtm.tfidf = removeSparseTerms(dtm.tfidf, 0.999)
  tfidf.matrix = as.matrix(dtm.tfidf)
  if(structKey == 'tfidf') {
    return(dtm.tfidf)
  }
}