Skip to main content

Run all regression and classification models with its default parameters

Project description

# run_models

run_models lets you run many regression and classification models with their default parameters and rank them by RMSE or AUC.

Installation

pip install run_models

Dependencies

numpy

pandas

scikit-learn

xgboost

lightgbm

catboost

matplotlib

How to use it?

run_models comes with two functions:

  • run_models.run_regressors(data, target_column) for regression tasks - All models are ranked by RMSE.
  • run_models.run_classifiers(data, target_column) for classification tasks - All models are ranked y AUC.

import run_models as rm

fig = rm.run_classifiers(titanic_data, 'Survived')

Project details


Download files

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

Files for run-models, version 0.0.4
Filename, size File type Python version Upload date Hashes
Filename, size run_models-0.0.4-py3-none-any.whl (4.1 kB) File type Wheel Python version py3 Upload date Hashes View hashes

Supported by

Elastic Elastic Search Pingdom Pingdom Monitoring Google Google BigQuery Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN DigiCert DigiCert EV certificate StatusPage StatusPage Status page