topsbm - A scikit-learn extension for Topic Models based on Stochastic Block Models

Martin Gerlach, Tiago P. Peixoto, and Eduardo G. Altmann, “A network approach to topic models,” Science Advances (2018)

Software ported to Scikit-learn format by the Sydney Informatics Hub at the University of Sydney.

• Installation
  • Installing dependencies
  • Check your installation
• API Documentation
• Example: Introduction to topsbm
  • Setup: Load a corpus
  • Fit the model
  • Plotting the graph and block structure
  • Topics
  • Topic-distribution in each document
  • Extra: Clustering of documents - for free.

• Index