diff --git a/README.md b/README.md index eab303ad..6b526aa5 100644 --- a/README.md +++ b/README.md @@ -3,12 +3,14 @@
![GitHub Actions Workflow Status](https://img.shields.io/github/actions/workflow/status/JamesYang007/adelie/test_docs.yml) +[![Downloads](https://static.pepy.tech/badge/adelie)](https://pepy.tech/project/adelie) [![PyPI Downloads](https://img.shields.io/pypi/dm/adelie.svg?label=PyPI%20downloads)](https://pypi.org/project/adelie/) ![versions](https://img.shields.io/pypi/pyversions/adelie.svg) ![PyPI - Version](https://img.shields.io/pypi/v/adelie) ![GitHub Release](https://img.shields.io/github/v/release/JamesYang007/adelie) -Adelie is a fast and flexible Python package for solving group lasso and elastic net problems. +Adelie is a fast and flexible Python package for solving +lasso, elastic net, group lasso, and group elastic net problems. - **Installation**: [https://jamesyang007.github.io/adelie/notebooks/installation.html](https://jamesyang007.github.io/adelie/notebooks/installation.html) - **Documentation**: [https://jamesyang007.github.io/adelie](https://jamesyang007.github.io/adelie/) diff --git a/adelie/diagnostic.py b/adelie/diagnostic.py index f027dd99..e7fc228f 100644 --- a/adelie/diagnostic.py +++ b/adelie/diagnostic.py @@ -196,6 +196,7 @@ def objective( y = glm.y if groups is None: + p = X.shape[1] groups = np.arange(p, dtype=int) if glm.is_multi: diff --git a/docs/sphinx/index.rst b/docs/sphinx/index.rst index 73ae33cd..cf44dcc2 100644 --- a/docs/sphinx/index.rst +++ b/docs/sphinx/index.rst @@ -17,7 +17,8 @@ Adelie documentation `Issue Tracker `_ -Adelie is a fast and flexible Python package for solving group lasso and elastic net problems. +Adelie is a fast and flexible Python package for solving +lasso, elastic net, group lasso, and group elastic net problems. It offers a general purpose group elastic net solver, a wide range of matrix classes that can exploit special structure to allow large-scale inputs, and an assortment of generalized linear model (GLM) classes for fitting various types of data.