Optuna-Integration is an integration module of Optuna. This package allows us to use Optuna, an automatic Hyperparameter optimization software framework, integrated with many useful tools like PyTorch, sklearn, TensorFlow, etc.
Optuna-Integration API reference is here.
- AllenNLP (example)
- Catalyst (example)
- Chainer (example)
- ChainerMN (example)
- Keras (example)
- MXNet (example)
- SHAP
- skorch (example)
- tf.keras (example)
Optuna-Integration is available at the Python Package Index and on Anaconda Cloud.
# PyPI
$ pip install optuna-integration
# Anaconda Cloud
$ conda install -c conda-forge optuna-integration
Optuna-Integration supports from Python 3.7 to Python 3.10.
Also, we also provide Optuna docker images on DockerHub.
- GitHub Discussions for questions.
- GitHub Issues for bug reports and feature requests.
Any contributions to Optuna-Integration are more than welcome!
For general guidelines how to contribute to the project, take a look at CONTRIBUTING.md.
Takuya Akiba, Shotaro Sano, Toshihiko Yanase, Takeru Ohta, and Masanori Koyama. 2019. Optuna: A Next-generation Hyperparameter Optimization Framework. In KDD (arXiv).