Method __repr__

src/ml_models/topic_modeling/topic_model.py:29–30

(self)

        self.config = config

    def __repr__(self) -> str:
        return f"TopicModel(config={self.config})"

    def fit_topic_model(self, topic_model, data, embeddings, key="query"):
        topics, probs = topic_model.fit_transform(data[key].to_list(), embeddings)