Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
def get_artm_topics(model_artm):
"""Return string representations of artm topics"""
topics = []
for topic_name in model_artm.topic_names:
topics.append(model_artm.score_tracker['TopTokensScore'].last_tokens[topic_name])
return topics
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment