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

x-tabdeveloping · web-flow · commit 0ec180e7d6e5 · 2024-06-06T12:54:33.000+02:00
fix ranking of negative documents
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]

Original file line number	Diff line number	Diff line change
`@@ -217,6 +217,7 @@ def _representative_docs(`
`217`	`217`	`lowest = lowest[`
`218`	`218`	`np.argsort(document_topic_matrix[lowest, topic_id])`
`219`	`219`	`]`
	`220`	`+ lowest = lowest[::-1]`
`220`	`221`	`scores = document_topic_matrix[lowest, topic_id]`
`221`	`222`	`for document_id, score in zip(lowest, scores):`
`222`	`223`	`doc = raw_documents[document_id]`