Classes for working with topic modelling algorithms. End-users should not need to use these classes directly.

Expand source code
"""
Classes for working with topic modelling algorithms.
End-users should not need to use these classes directly.
"""

from .lda import LDAModel
from .hdp import HDPModel

# TODO: Keep a central registry of model types

Sub-modules

ignis.models.base

Base class for all ignis models. Should never be instantiated directly.

ignis.models.hdp
ignis.models.lda

Topic modelling using a tomotopy LDA model.