scikit-learn / scikit-learn/scikit-learn
Analyze the practical relevance of GBT hyperparameters in the accuracy / training speed tradeoff
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 67.3k
- Forks
- 27.4k
- Avg merge
- 1d 15h
- Merged PRs (30d)
- 58
Description
The following paper ran many hyper-parameter tuning experiments.
The raw CSV data is available for download here:
We could try to mine this data to find hyper-params configuration implemented in xgboost or lightgbm that:
- allow xgboost or lightgbm or catboost to be on the Pareto optimal frontier of the predictive/computational performance tradeoff;
- have no equivalent in scikit-learn.
This would help identify which missing entries of the table in #27873 have the most user-facing impact, and possibly also identify features of xgboost that are not implemented in scikit-learn at all while being very relevant to reach good predictive performance.
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start by reading the linked paper and inspecting the raw CSV files in LeoGrin/tabular-benchmark/analyses/results. Compare the reported XGBoost, LightGBM, and CatBoost configurations with the table in #27873 and scikit-learn equivalents. Done means identifying configurations on the predictive/computational Pareto frontier and documenting which have no meaningful scikit-learn equivalent.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- data, machine-learning
- Issue type
- Feature
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Quiet
- Clarity
- Needs clarification
- Newbie friendliness
- 38/100