Fugue

A Topic Modeling Package

Fugue is a research oriented topic modeling package for text mining and machine learning practitioners, designed for understanding models and algorithms. The philosophy of the package is to include mature models and algorithms that represent the state-of-the-art of research in topic modeling.

Models/Algorithms

• Latent Dirichlet allocation (LDA) with collapsed Gibbs sampling
  • Using "Estimate theta" method for computing perplexity in test documents.
  • Using multiple MCMC chains to average results.
  • Slice sampling for hyper-parameter sampling
  • Optimizain methods for hyper-parameter tuning

Roadmap

For 1.0, See Detailed Plan.

Prerequisites

• Gradle
• JDK
• Python 3.7

The design of the package is to limit the prerequesite packages and therefore you can easily experiment new models with minimum environments.

Platforms

• Mac/Linux

How to Compile

$ python Fugue.py

Run AP example

$ python Fugue.py --task train

License

Licensed under an Apache-2.0 license.