Update src/neurostatslib/solver.py

Co-authored-by: William F. Broderick <billbrod@gmail.com>
flatironinstitute · Nov 1, 2023 · 881d503 · 881d503
1 parent dbdca86
commit 881d503
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/neurostatslib/solver.py b/src/neurostatslib/solver.py
@@ -541,7 +541,7 @@ class GroupLassoSolver(ProxGradientSolver):
     Attributes
     ----------
     mask : Union[jnp.ndarray, NDArray]
-        A mask array indicating groups of features for regularization.
+        A 2d mask array indicating groups of features for regularization.
         Each row represents a group of features.
         Each column corresponds to a feature, where a value of 1 indicates that the feature belongs
         to the group, and a value of 0 indicates it doesn't.