Skip to main content

A Python library for tuning machine learning models.

Project description



Downloads PyPI License DOI

The model_tuner library is a versatile and powerful tool designed to facilitate the training, evaluation, and tuning of machine learning models. It supports various functionalities such as handling imbalanced data, applying different scaling and imputation techniques, calibrating models, and conducting cross-validation. This library is particularly useful for model selection, hyperparameter tuning, and ensuring optimal performance across different metrics.

Prerequisites

Before installing model_tuner, ensure your system meets the following requirements:

Python Version

model_tuner requires Python 3.7 or higher. Specific dependency versions vary depending on your Python version.

Dependencies

The following dependencies will be automatically installed when you install model_tuner via pip:

For Python 3.7:

  • joblib==1.3.2
  • tqdm==4.66.4
  • catboost==1.2.7
  • numpy==1.21.4
  • pandas==1.1.5
  • scikit-learn==0.23.2
  • scipy==1.4.1
  • imbalanced-learn==0.7.0
  • scikit-optimize==0.8.1
  • xgboost==1.6.2

For Python 3.8 to 3.10:

  • joblib==1.3.2
  • tqdm==4.66.4
  • catboost==1.2.7
  • setuptools==75.1.0
  • wheel==0.44.0
  • numpy>=1.19.5, <2.0.0
  • pandas>=1.3.5, <2.2.3
  • scikit-learn>=1.0.2, <1.4.0
  • scipy>=1.6.3, <1.11
  • imbalanced-learn==0.12.4
  • scikit-optimize==0.10.2
  • xgboost==2.1.2

For Python 3.11 and higher:

  • joblib==1.3.2
  • tqdm==4.66.4
  • catboost==1.2.7
  • setuptools==75.1.0
  • wheel==0.44.0
  • numpy>=1.19.5, <2.0.0
  • pandas>=1.3.5, <2.2.2
  • scikit-learn==1.5.1
  • scipy==1.14.0
  • imbalanced-learn==0.12.4
  • scikit-optimize==0.10.2
  • xgboost==2.1.2

💾 Installation

You can install model_tuner directly from PyPI:

pip install model_tuner

📄 Official Documentation

https://uclamii.github.io/model_tuner

🌐 Author Website

https://www.mii.ucla.edu/

⚖️ License

model_tuner is distributed under the Apache License. See LICENSE for more information.

📚 Citing model_tuner

If you use model_tuner in your research or projects, please consider citing it.

@software{funnell_2024_12727322,
  author       = {Funnell, Arthur and
                  Shpaner, Leonid and
                  Petousis, Panayiotis},
  title        = {Model Tuner},
  month        = jul,
  year         = 2024,
  publisher    = {Zenodo},
  version      = {0.0.26b},
  doi          = {10.5281/zenodo.12727322},
  url          = {https://doi.org/10.5281/zenodo.12727322}
}

Support

If you have any questions or issues with model_tuner, please open an issue on this GitHub repository.

Acknowledgements

This work was supported by the UCLA Medical Informatics Institute (MII) and the Clinical and Translational Science Institute (CTSI). Special thanks to Dr. Alex Bui for his invaluable guidance and support, and to Panayiotis Petousis for his original contributions to this codebase.

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

model_tuner-0.0.26b0.tar.gz (33.3 kB view details)

Uploaded Source

Built Distribution

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

model_tuner-0.0.26b0-py3-none-any.whl (33.0 kB view details)

Uploaded Python 3

File details

Details for the file model_tuner-0.0.26b0.tar.gz.

File metadata

  • Download URL: model_tuner-0.0.26b0.tar.gz
  • Upload date:
  • Size: 33.3 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.13.1

File hashes

Hashes for model_tuner-0.0.26b0.tar.gz
Algorithm Hash digest
SHA256 23c73f014a65bc6739a1b92db3058fd0dc3f23054d217bef0538de2f7b80634e
MD5 4ce9d5cbf54623ce93688a8739386000
BLAKE2b-256 9a2e7a66947965b1074f2e38e4352a5200abefb4a9ae8d670cd3f8a741c5ea05

See more details on using hashes here.

File details

Details for the file model_tuner-0.0.26b0-py3-none-any.whl.

File metadata

  • Download URL: model_tuner-0.0.26b0-py3-none-any.whl
  • Upload date:
  • Size: 33.0 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.13.1

File hashes

Hashes for model_tuner-0.0.26b0-py3-none-any.whl
Algorithm Hash digest
SHA256 26df50c2c53096c3385ff24b145403cb9b29d9e2c33e4f4d7cb57e38019b13e2
MD5 770d3fefa8e162a2f7984be6f7f5769d
BLAKE2b-256 5b1002b0c51790301948244c688fddf14300b3a3872ea4b0e0a7245b4a02c1b9

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