search

x-tabdeveloping / turftopic

Robust and fast topic models with sentence-transformers.
https://x-tabdeveloping.github.io/turftopic/
MIT License
13 stars 4 forks source link

Hierarchical Topic Modeling (Divisive) #56

Closed x-tabdeveloping closed 1 month ago

x-tabdeveloping commented 1 month ago

I added Hierarchical modeling to KeyNMF in a divisive manner. In short, subtopics are calculated with a special case of weighted NMF, where the document weights are determined by the importance of the parent topic in the document. I also added pretty printing and plotting utilities for topic hierarchies.

I also prepared the package for a new release encompassing fastopic as well.

x-tabdeveloping commented 1 month ago

Yeah adding a test is probably not a bad idea