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.
Before you install model_tuner
, ensure your system meets the following requirements:
Python
: Version3.7
or higher is required to runmodel_tuner
.
Additionally, model_tuner
depends on the following packages, which will be automatically installed when you install model_tuner
using pip:
-
numpy
: version1.21.6
or higher -
pandas
: version1.3.5
or higher -
joblib
: version1.3.2
or higher -
scikit-learn
: version1.0.2
or higher -
scipy
: version1.7.3
or higher -
tqdm
: version4.66.4
or higher
You can install model_tuner
directly from PyPI:
pip install model_tuner
https://uclamii.github.io/model_tuner
model_tuner
is distributed under the Apache License. See LICENSE for more information.
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.19a},
doi = {10.5281/zenodo.12727322},
url = {https://doi.org/10.5281/zenodo.12727322}
}
If you have any questions or issues with model_tuner
, please open an issue on this GitHub repository.
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.