Search code examples
mahoutldatopic-modeling

Run cvb in mahout 0.8


The current Mahout 0.8-SNAPSHOT includes a Collapsed Variational Bayes (cvb) version for Topic Modeling and removed the Latent Dirichlet Analysis (lda) approach, because cvb can be parallelized way better. Unfortunately there is only documentation for lda on how to run an example and generate meaningful output.

Thus, I want to:

  • preprocess some texts correctly
  • run the cvb0_local version of cvb
  • inspect the results by looking at the top n words in each of the generated topics

Solution

  • So here are the subsequent Mahout commands I had to call in a linux shell to do it. $MAHOUT_HOME points to my mahout/bin folder.

    $MAHOUT_HOME/mahout seqdirectory \
        -i path/to/directory/with/texts \
        -o out/sequenced
    
    $MAHOUT_HOME/mahout seq2sparse -i out/sequenced \
        -o out/sparseVectors \
        --namedVector \
        -wt tf
    
    $MAHOUT_HOME/mahout rowid \
        -i out/sparseVectors/tf-vectors/ \
        -o out/matrix
    
    $MAHOUT_HOME/mahout cvb0_local \
        -i out/matrix/matrix \
        -d out/sparseVectors/dictionary.file-0 \
        -a 0.5 \
        -top 4 -do out/cvb/do_out \
        -to out/cvb/to_out
    

    Inspect the output by showing the top 10 words of each topic:

    $MAHOUT_HOME/mahout vectordump \
        -i out/cvb/to_out \
        --dictionary out/sparseVectors/dictionary.file-0 \
        --dictionaryType sequencefile \
        --vectorSize 10 \
        -sort out/cvb/to_out