From 93dcc71c0cfe4934986be5fe6bcd8de643e6af4f Mon Sep 17 00:00:00 2001 From: maaz Date: Thu, 27 Jul 2023 18:28:57 -0500 Subject: [PATCH] Skmultilearn binary relevance produces sparse output. Handling that sparse output --- lime/lime_text.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/lime/lime_text.py b/lime/lime_text.py index 7055a9b59..d40238b0d 100644 --- a/lime/lime_text.py +++ b/lime/lime_text.py @@ -481,5 +481,10 @@ def distance_fn(x): data[i, inactive] = 0 inverse_data.append(indexed_string.inverse_removing(inactive)) labels = classifier_fn(inverse_data) + + # if labels are a sparse matrix, convert to a dense array + if sp.sparse.issparse(labels): + labels = labels.toarray() + distances = distance_fn(sp.sparse.csr_matrix(data)) return data, labels, distances