diff --git a/README.md b/README.md index 1272ed2..986f048 100644 --- a/README.md +++ b/README.md @@ -12,7 +12,7 @@ where the L0 norm counts the number of nonzeros in the coefficients vector B. He ## Installation The toolkit is implemented in Python 3. To install it, run the following command: ```bash -pip install L0BnB +pip install l0bnb ``` ## A Quick Start in Python @@ -43,8 +43,8 @@ sols is a list of solutions, each corresponding to a different lambda_0. Below we inspect the solution with index 4. The estimated coefficients vector "b_estimated" and the intercept term can be accessed as follows: """ -b_estimated = sol[4]["B"] # a numpy array. -intercept = sol[4]["B0"] +b_estimated = sols[4]["B"] # a numpy array. +intercept = sols[4]["B0"] # To check the nonzero indices in b_estimated: print("Nonzero indices in b_estimated: ", np.nonzero(b_estimated)[0]) diff --git a/setup.py b/setup.py index 2d9d996..636f62f 100644 --- a/setup.py +++ b/setup.py @@ -15,7 +15,7 @@ def readme(): author_email="alikassemsaab@gmail.com", url='https://github.com/alisaab/l0bnb', download_url="https://github.com/alisaab/l0bnb/archive/0.0.5.tar.gz", - install_requires=["numpy >= 1.18.2", "scipy >= 1.4.1", "numba >= 0.48.0"], + install_requires=["numpy >= 1.18.1", "scipy >= 1.4.1", "numba >= 0.48.0"], version="0.0.5", packages=find_packages(), include_package_data=True,