ControlBurn v0.1.3
This package implements ControlBurn in python. ControlBurn is a feature selection algorithm that uses a weighted LASSO-based feature selection to prune unnecessary features from tree ensembles. The algorithm is efficient and only requires a single training iteration to run.
Tree ensembles distribute feature importance scores evenly amongst groups of correlated features. The average feature ranking of the correlated group is suppressed, which reduces interpretability and complicates feature selection. Like the linear LASSO, ControlBurn assigns all the feature importance of a correlated group of features to a single feature. The algorithm is able to quickly select a subset of important independent features for further analysis.
Installation
The easiest way to install ControlBurn is through pip.
!pip install ControlBurn==0.1.3
Dependencies
ControlBurn works on python 3.7 or above. The following packages are required.
- numpy (1.20.1)
- pandas (1.2.4)
- sklearn (0.24.1)
- mosek (9.2.47)
- cvxpy (1.1.13)
Quick Start
from ControlBurn.ControlBurnModel import ControlBurnClassifier
cb = ControlBurnClassifier(alpha = 0.1)
cb.fit(X,y)
print(cb.features_selected_) #print selected features
print(cb.feature_importances_) #print feature importances
pred = cb.predict(X) #return predictions of polished model using selected features
Reference Paper
ControlBurn: Feature Selection by Sparse Forests B. Liu, M. Xie, and M. Udell
ACM SIGKDD Conference on Knowledge Discovery and Data Mining (KDD), 2021
1 maintainer
Metadata
1 maintainer
Release files for ControlBurn 0.1.3
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | Provenance | |
|---|---|---|---|---|
| ControlBurn-0.1.3.tar.gz | 16.8 kB | Without provenance | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| ControlBurn-0.1.3-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 33.6 kB
Release files / ControlBurn-0.1.3.tar.gz
| Download URL | ControlBurn-0.1.3.tar.gz |
|---|---|
| Size | 16.8 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
ad5fe989acebc025ac58496cb4b5f8def20ba5fa90a0d48d9dcee2358aa9324b
|
|
BLAKE2b-256 checksum How to use checksums |
7d1a7d10a29bff6cadc33ed3159b9136003a197be96501120e86cab240e29909
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/3.7.1 importlib_metadata/4.10.1 pkginfo/1.8.2 requests/2.27.1 requests-toolbelt/0.9.1 tqdm/4.62.3 CPython/3.9.13
|
Release files / ControlBurn-0.1.3-py3-none-any.whl
| Download URL | ControlBurn-0.1.3-py3-none-any.whl |
|---|---|
| Size | 16.8 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
12d6594091795b3f30f9f5509337081babe999e1944404ab4968f12c4dec183e
|
|
BLAKE2b-256 checksum How to use checksums |
7dc426a6b88cb5ef4f9c9a2bef40e6453cee5025b541ef8158a6791805c187ed
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/3.7.1 importlib_metadata/4.10.1 pkginfo/1.8.2 requests/2.27.1 requests-toolbelt/0.9.1 tqdm/4.62.3 CPython/3.9.13
|
Security for ControlBurn 0.1.3
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
No provenance found
None of the files in this release were uploaded with provenance metadata.
A note on package security
PyPI is a repository for distributing packages, not a service for auditing them. While third-party security researchers and our staff do their best to detect and remove malicious packages, PyPI cannot guarantee the quality, security or suitability of any given package.
To learn more about protecting yourself from malicious actors, visit our package security documentation.
