HyPSTER - HyperParameter optimization on STERoids
You have just found HyPSTER.
HyPSTER is a brand new Python package built on top of Optuna.
It helps you find compact and accurate ML Pipelines while staying light and efficient.
HyPSTER uses state of the art algorithms for sampling hyperparameters (e.g. TPE, CMA-ES) and pruning unpromising trials (e.g. Asynchronous Successive Halving), combined with cross-validated early stopping and adaptive learning rates, all packed up in a simple sklearn API that allows for automatic Preprocessing pipeline selection and supports your favorite ML packages (e.g. XGBoost, LightGBM, CatBoost, SGDClassifier) out of the box.
And yes, it supports multi CPU/GPU training.
Guiding principles
User friendliness
Modularity
Easy extensibility
Work with Python
Getting started
from hypster import HyPSTERClassifier
frameworks = ["xgboost", "lightgbm", "sklearn"]
model_types = ["linear", "tree-based"]
clf = HyPSTERClassifier(frameworks, model_types,
scoring="roc_auc", max_iter=30,
n_jobs=-1, random_state=SEED)
clf.fit(X_train, y_train, cat_cols=cat_cols, n_trials=30)
clf.predict_proba(X_test)
Installation
> pip install hypster