Cherrypick - ML model Orchestration

Cherrypick is an automated machine learning (Automated-ML) library that simplifies:

  • Data preprocessing

  • Model selection

  • Outlier removal

  • Explainability

It helps you quickly train, evaluate, and interpret models with minimal effort.

Quick Example

from cherrypick import Orchestrator

# For Regression:
cherry = Orchestrator(
                    problem_Statement = 'regression',
                    train = train_reg,
                    test = test_reg,
                    focus_regressor = 'mse',
                    file_dir = 'model_'
                    )
# Orchestrates the model selection
cherry.orchestrate()

# For Classification
  cherry = Orchestrator(
                      problem_Statement = 'classification',
                      train = train_classify,
                      test = test_classify,
                      focus_classifier = 'f1score',
                      file_dir = 'model_'
                    )
# Orchestrates the model selection
cherry.orchestrate()

Key Features

  • Automated model selection

  • Built-in explainability (SHAP)

  • Performance evaluation tools

  • Data preprocessing pipeline

Installation

Install via pip:

pip install cherrypick-ml