Merge pull request #42 from rbroc/fix-neg-doc-ranking

fix ranking of negative documents
x-tabdeveloping · Jun 6, 2024 · 0ec180e · 0ec180e
2 parents 89a96a5 + 26e56c9
commit 0ec180e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/turftopic/base.py b/turftopic/base.py
@@ -217,6 +217,7 @@ def _representative_docs(
             lowest = lowest[
                 np.argsort(document_topic_matrix[lowest, topic_id])
             ]
+            lowest = lowest[::-1]
             scores = document_topic_matrix[lowest, topic_id]
             for document_id, score in zip(lowest, scores):
                 doc = raw_documents[document_id]