tommy.model.topic_model

Classes

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