Add BERTopic.

2025-08-12 19:01:20 +08:00
parent e2323d579c
commit c5c530775e
256 changed files with 28666 additions and 0 deletions
@@ -0,0 +1,8 @@
+import copy
+import pytest
+
+
+@pytest.mark.parametrize("model", [("kmeans_pca_topic_model"), ("base_topic_model"), ("custom_topic_model")])
+def test_term_rank(model, request):
+    topic_model = copy.deepcopy(request.getfixturevalue(model))
+    topic_model.visualize_term_rank()