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 you install model_tuner, ensure your system meets the following requirements:

  • Python: Version 3.7 or higher is required to run model_tuner.

Additionally, model_tuner depends on the following packages, which will be automatically installed when you install model_tuner using pip:

  • numpy: version 1.21.6 or higher

  • pandas: version 1.3.5 or higher

  • joblib: version 1.3.2 or higher

  • scikit-learn: version 1.0.2 or higher

  • scipy: version 1.7.3 or higher

  • tqdm: version 4.66.4 or higher

💾 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.16a},
  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.16a0.tar.gz (24.6 kB view hashes)

Uploaded Source

Built Distribution

model_tuner-0.0.16a0-py3-none-any.whl (23.7 kB view hashes)

Uploaded Python 3

Supported by

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