topk
High-performance KNN similarity functions in Python, optimized for sparse matrices
[ICML2022] Stochastic smoothing of the top-K calibrated hinge loss for deep imbalanced classification
