Skip to main content

A hyperparameter tuner for XGBoost.

Project description

xgbtuner

A world-class, easy-to-use hyperparameter tuner for XGBoost models, built on top of scikit-learn’s GridSearchCV and RandomizedSearchCV. Supports both classification and regression out of the box, with robust error handling, comprehensive testing, and flexible customization.

Demo Usage: https://colab.research.google.com/drive/1eSbeIKlVeUo-0_PLUtr0lBi3WxlBG9st#scrollTo=yybTYeBVr5Nh


Features

  • Grid Search & Random Search
    Choose between exhaustive grid search or randomized search for hyperparameter exploration.
  • Classification & Regression
    One API for both objectives—simply set objective="classification" or "regression".
  • Default & Custom Grids
    Sensible default parameter grid, plus ability to pass your own param_grid.
  • Robust Input Validation
    Checks for array-like inputs, consistent lengths, and raises clear errors.
  • Detailed Logging
    Built-in logging statements to trace tuning progress and errors.
  • Fully Tested
    Over 10 unit tests cover edge cases, custom grids, list inputs, invalid configs, and more.
  • Scikit-Learn Compatible
    Behaves like any estimator: .tune(), .predict(), and .predict_proba() for classification.

Installation

Install from PyPI:

pip install xgboost-tuner-pack

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

xgboost_tuner_pack-0.1.5.tar.gz (7.2 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

xgboost_tuner_pack-0.1.5-py3-none-any.whl (6.9 kB view details)

Uploaded Python 3

File details

Details for the file xgboost_tuner_pack-0.1.5.tar.gz.

File metadata

  • Download URL: xgboost_tuner_pack-0.1.5.tar.gz
  • Upload date:
  • Size: 7.2 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.13.2

File hashes

Hashes for xgboost_tuner_pack-0.1.5.tar.gz
Algorithm Hash digest
SHA256 bf3979f4d61e51a76657ffa98baa17dbe826bc05de2a4969cd35b0762df1cfc6
MD5 baebcb42462b893ab1287e200d005491
BLAKE2b-256 11e301e02880f2ab620bece107d4c479b2d3429d9430299f69b8d2564f65ecc5

See more details on using hashes here.

File details

Details for the file xgboost_tuner_pack-0.1.5-py3-none-any.whl.

File metadata

File hashes

Hashes for xgboost_tuner_pack-0.1.5-py3-none-any.whl
Algorithm Hash digest
SHA256 37ca5c95f223eda5d1ef948394f3689de2813efb94c34c9688d9f7fcba73a64c
MD5 94e16921a0b36f70ed409b41e22010fe
BLAKE2b-256 0d18d0ff44aa9856b0d12728e2c964541716f73b1eb1c14fbf4ac047499fe76f

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page