tommy.model.topic_model.TopicModel

class tommy.model.topic_model.TopicModel(model: LdaModel | Nmf | BERTopic = None, dictionary: Dictionary = None, corpus: list[list[tuple[int, int]]] = None, document_topics: list[tuple[Metadata, list[float]]] = None, used_corpus_version_id: int = None)[source]

Bases: object

dataclass that holds the data and objects necessary to run a topic modelling algorithm

corpus: list[list[tuple[int, int]]] = None
dictionary: Dictionary = None
document_topics: list[tuple[Metadata, list[float]]] = None
model: LdaModel | Nmf | BERTopic = None
used_corpus_version_id: int = None