Skip to content

Commit

Permalink
Added .merge_topics() to manually merge topics
Browse files Browse the repository at this point in the history
  • Loading branch information
MaartenGr committed Jul 4, 2022
1 parent 8b5084b commit 08129c8
Show file tree
Hide file tree
Showing 6 changed files with 189 additions and 90 deletions.
1 change: 1 addition & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -215,6 +215,7 @@ For quick access to common functions, here is an overview of BERTopic's main met
| Update topic representation | `.update_topics(docs, topics, n_gram_range=(1, 3))` |
| Generate topic labels | `.generate_topic_labels()` |
| Set topic labels | `.set_topic_labels(my_custom_labels)` |
| Merge topics | `.merge_topics(docs, topics, topics_to_merge)` |
| Reduce nr of topics | `.reduce_topics(docs, topics, nr_topics=30)` |
| Find topics | `.find_topics("vehicle")` |
| Save model | `.save("my_model")` |
Expand Down

0 comments on commit 08129c8

Please sign in to comment.