From 881d503d48d5c15b57a616fbcf0a5407dc9c1e7e Mon Sep 17 00:00:00 2001 From: Edoardo Balzani Date: Wed, 1 Nov 2023 12:15:10 -0400 Subject: [PATCH] Update src/neurostatslib/solver.py Co-authored-by: William F. Broderick --- src/neurostatslib/solver.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/neurostatslib/solver.py b/src/neurostatslib/solver.py index 178c9770..05b043f5 100644 --- 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.