scikit-learn / scikit-learn/scikit-learn

Analyze the practical relevance of GBT hyperparameters in the accuracy / training speed tradeoff

Open
#33,137 5 comments 3 reactions 0 assignees View on GitHub

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

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. 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

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.