From f38cb47b359b8fa128b0eae62e244756cd494e42 Mon Sep 17 00:00:00 2001 From: myui Date: Tue, 10 Dec 2024 15:42:16 +0900 Subject: [PATCH] Revised setcol() for csr --- rtrec/models/internal/slim_elastic.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/rtrec/models/internal/slim_elastic.py b/rtrec/models/internal/slim_elastic.py index cdd82d4..99e38d6 100644 --- a/rtrec/models/internal/slim_elastic.py +++ b/rtrec/models/internal/slim_elastic.py @@ -96,7 +96,9 @@ def set_col(self, j: int, values: ArrayLike) -> None: values : ArrayLike The new values for the column. """ - self.csc_matrix[:, j] = values + start, end = self.csc_matrix.indptr[j], self.csc_matrix.indptr[j+1] + assert len(values) == end - start, f"Values must have the same length as the column: {len(values)} != {end - start}" + self.csc_matrix.data[start:end] = values class FeatureSelectionWrapper: