Files
bettafish-company/LLMTopicDetection_BERTopic/tests/test_plotting/test_term_rank.py
T
戒酒的李白 c5c530775e Add BERTopic.
2025-08-12 19:01:20 +08:00

9 lines
276 B
Python

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()