explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-09-18 18:12:45 +03:00
Code Issues Packages Projects Releases Wiki Activity

Change topk to sort descending

Browse Source 
Shouldn't change correctness but is a little clearer
This commit is contained in:
Paul O'Leary McCann 2021-06-13 19:42:24 +09:00
parent d71198ed36
commit 96be7e8858
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
spacy/ml/models/coref_util.py
View File

@ -34,7 +34,7 @@ def topk(xp, arr, k, axis=None):
vals = xp.take_along_axis(arr, idxs, axis=1) vals = xp.take_along_axis(arr, idxs, axis=1)
sidxs = xp.argsort(vals, axis=1) sidxs = xp.argsort(-vals, axis=1)
# map these idxs back to the original # map these idxs back to the original
oidxs = xp.take_along_axis(idxs, sidxs, axis=1) oidxs = xp.take_along_axis(idxs, sidxs, axis=1)
svals = xp.take_along_axis(vals, sidxs, axis=1) svals = xp.take_along_axis(vals, sidxs, axis=1)