No project description provided
Project description
fseval
A Feature Selector and Feature Ranker benchmarking library. Neatly integrates with wandb and sklearn. Uses Hydra as a config parser.
Usage
pip install fseval
fseval help:
fseval --help
Now, create a wandb account and login to the CLI. We are now able to run benchmarks 💪🏻. The results will automatically be uploaded to the wandb dashboard.
Run ReliefF on Iris dataset:
fseval dataset=iris estimator@pipeline.ranker=anova_f_value
Supported Feature Rankers
A collection of feature rankers are already built-in, which can be used without further configuring. Others need their dependencies installed. List of rankers:
Ranker | Dependency | Command line argument |
---|---|---|
ANOVA F-Value | <no dep> | estimator@pipeline.ranker=anova_f_value |
Boruta | pip install Boruta |
estimator@pipeline.ranker=boruta |
Chi2 | <no dep> | estimator@pipeline.ranker=chi2 |
Decision Tree | <no dep> | estimator@pipeline.ranker=decision_tree |
FeatBoost | pip install git+https://github.com/dunnkers/FeatBoost.git@support-cloning (ℹ️) |
estimator@pipeline.ranker=featboost |
MultiSURF | pip install skrebate |
estimator@pipeline.ranker=multisurf |
Mutual Info | <no dep> | estimator@pipeline.ranker=mutual_info |
ReliefF | pip install skrebate |
estimator@pipeline.ranker=relieff |
Stability Selection | pip install git+https://github.com/dunnkers/stability-selection.git@master matplotlib (ℹ️) |
estimator@pipeline.ranker=stability_selection |
TabNet | pip install pytorch-tabnet |
estimator@pipeline.ranker=tabnet |
XGBoost | pip install xgboost |
estimator@pipeline.ranker=xgb |
ℹ️ This library was customized to make it compatible with the fseval pipeline.
If you would like to install simply all dependencies, download the fseval requirements.txt file and run pip install -r requirements.txt
.
About
Built by Jeroen Overschie as part of the Masters Thesis (Data Science and Computational Complexity track at the University of Groningen).
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.