Optunity is a library containing various optimizers for hyperparameter tuning. Hyperparameter tuning is a recurrent problem in many machine learning tasks, both supervised and unsupervised.This package provides several distinct approaches to solve such problems including some helpful facilities such as cross-validation and a plethora of score functions.
From an optimization point of view, the tuning problem can be considered as follows: the objective function is non-convex, non-smooth and typically expensive to evaluate. Tuning examples include optimizing regularization or kernel parameters.
The Optunity library is implemented in Python and allows straightforward integration in other machine learning environments. Optunity is currently also supported in /wrappers/r, MATLAB, GNU Octave and Java through Jython.
Optunity is free software, using a BSD-style license.
As a simple example of Optunity’s features, the code below demonstrates how to tune an SVM with RBF kernel using Optunity and scikit-learn. This involves optimizing the hyperparameters gamma and C:
import optunity import optunity.metrics import sklearn.svm # score function: twice iterated 10-fold cross-validated accuracy @optunity.cross_validated(x=data, y=labels, num_folds=10, num_iter=2) def svm_auc(x_train, y_train, x_test, y_test, C, gamma): model = sklearn.svm.SVC(C=C, gamma=gamma).fit(x_train, y_train) decision_values = model.decision_function(x_test) return optunity.metrics.roc_auc(y_test, decision_values) # perform tuning optimal_pars, _, _ = optunity.maximize(svm_auc, num_evals=200, C=[0, 10], gamma=[0, 1]) # train model on the full training set with tuned hyperparameters optimal_model = sklearn.svm.SVC(**optimal_pars).fit(data, labels)
For more examples, please see our Examples.
Issue the following commands to get started on Linux:
git clone https://github.com/claesenm/optunity.git export PYTHONPATH=$PYTHONPATH:$(pwd)/optunity/
Afterwards, importing optunity should work in Python:
python -c 'import optunity'
For a proper installation, run the following:
python optunity/setup.py install
or, if you have pip:
pip install optunity
Installation may require superuser priviliges.
Optunity is developed at the STADIUS lab of the dept. of electrical engineering at KU Leuven (ESAT). The main contributors to Optunity are:
- Python package
- framework design & implementation
- solver implementation
- communication protocol design & implementation
- MATLAB wrapper
- Octave wrapper
- communication protocol design
- R wrapper
- code examples