From 802cba4fbb8f3b9622002b9315fc450952f45866 Mon Sep 17 00:00:00 2001 From: Ji Ma <11808231+ma-ji@users.noreply.github.com> Date: Fri, 17 Dec 2021 09:47:11 -0600 Subject: [PATCH] add wordcloud font add `font_path` to `generate_topic_wordcloud` so that more fonts can be supported. otherwise, non-alphabet language can not display. --- top2vec/Top2Vec.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/top2vec/Top2Vec.py b/top2vec/Top2Vec.py index 8d0b17e..d63df72 100644 --- a/top2vec/Top2Vec.py +++ b/top2vec/Top2Vec.py @@ -2373,7 +2373,7 @@ def search_documents_by_documents(self, doc_ids, num_docs, doc_ids_neg=None, ret else: return doc_scores, doc_ids - def generate_topic_wordcloud(self, topic_num, background_color="black", reduced=False): + def generate_topic_wordcloud(self, topic_num, background_color="black", font_path=None, reduced=False): """ Create a word cloud for a topic. @@ -2419,5 +2419,6 @@ def generate_topic_wordcloud(self, topic_num, background_color="black", reduced= plt.imshow( WordCloud(width=1600, height=400, + font_path=font_path, background_color=background_color).generate_from_frequencies(word_score_dict)) plt.title("Topic " + str(topic_num), loc='left', fontsize=25, pad=20)