From df1ea34bbc68788aa73b08fd8999e9398a0e9f07 Mon Sep 17 00:00:00 2001 From: Petr Andriushchenko Date: Thu, 27 Apr 2023 14:27:08 +0200 Subject: [PATCH] simplify lambda function --- ammico/multimodal_search.py | 10 ++-------- 1 file changed, 2 insertions(+), 8 deletions(-) diff --git a/ammico/multimodal_search.py b/ammico/multimodal_search.py index de7c208..98aad39 100644 --- a/ammico/multimodal_search.py +++ b/ammico/multimodal_search.py @@ -318,14 +318,8 @@ class MultimodalSearch(AnalysisMethod): ) similarity = features_image_stacked @ multi_features_stacked.t() - sorted_lists = [ - sorted( - range(len(similarity)), - key=lambda k, value=i: similarity[k, value], - reverse=True, - ) - for i in range(len(similarity[0])) - ] + # similarity_soft_max = torch.nn.Softmax(dim=0)(similarity / 0.01) + sorted_lists = torch.argsort(similarity, dim=0, descending=True).T.tolist() places = [[item.index(i) for i in range(len(item))] for item in sorted_lists] for q in range(len(search_query)):