From 5b0991e4949b42cff3dc6d5fcf225210ba1ee97e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Nguy=E1=BB=85n=20Th=C6=B0=20Uy=C3=AAn?= Date: Sun, 29 Mar 2026 01:15:51 +0700 Subject: [PATCH] fix sorting order in aspect score for Companion, ComparERSub --- cornac/models/companion/recom_companion.pyx | 2 +- cornac/models/comparer/recom_comparer_sub.pyx | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/cornac/models/companion/recom_companion.pyx b/cornac/models/companion/recom_companion.pyx index 1678a8b1..2246bf4d 100644 --- a/cornac/models/companion/recom_companion.pyx +++ b/cornac/models/companion/recom_companion.pyx @@ -1026,7 +1026,7 @@ class Companion(Recommender): np.repeat(range(n_items), n_top_aspects).reshape( n_items, n_top_aspects ), - ts3[:, :-1].argsort(axis=1)[::-1][:, :n_top_aspects], + ts3[:, :-1].argsort(axis=1)[:, ::-1][:, :n_top_aspects], ] item_scores = ( self.alpha * top_aspect_scores.mean(axis=1) + (1 - self.alpha) * ts3[:, -1] diff --git a/cornac/models/comparer/recom_comparer_sub.pyx b/cornac/models/comparer/recom_comparer_sub.pyx index ec1173db..9b2bf661 100644 --- a/cornac/models/comparer/recom_comparer_sub.pyx +++ b/cornac/models/comparer/recom_comparer_sub.pyx @@ -769,7 +769,7 @@ class ComparERSub(MTER): np.repeat(range(self.num_items), n_top_aspects).reshape( self.num_items, n_top_aspects ), - ts3[:, :-1].argsort(axis=1)[::-1][:, :n_top_aspects], + ts3[:, :-1].argsort(axis=1)[:, ::-1][:, :n_top_aspects], ] known_item_scores = ( self.alpha * top_aspect_scores.mean(axis=1) + (1 - self.alpha) * ts3[:, -1]