Optuna-Integration

This package is an integration module of Optuna, an automatic Hyperparameter optimization software framework. The modules in this package provide users with extended functionalities for Optuna in combination with third-party libraries such as PyTorch, sklearn, and TensorFlow.

[!NOTE] You can find more information in our official documentations and API reference.

Installation

Optuna-Integration is available via pip and on conda.

# PyPI
$ pip install optuna-integration

# Anaconda Cloud
$ conda install -c conda-forge optuna-integration

[!IMPORTANT] As dependencies of all the modules are large and complicated, the commands above install only the common dependencies. Dependencies for each module can be installed via pip. For example, if you would like to install the dependencies of optuna_integration.botorch and optuna_integration.lightgbm, you can install them via:
$ pip install optuna-integration[botorch,lightgbm]

[!NOTE] Optuna-Integration supports from Python 3.9 to Python 3.13. Optuna Docker image is also provided at DockerHub.

Integration Modules

Here is the table of optuna-integration modules:

Third Party Library	Example
BoTorch	Unavailable
CatBoost	CatBoostPruningCallback
Dask	DaskStorage
FastAI	FastAIPruningCallback
Keras	KerasPruningCallback
LightGBM	LightGBMPruningCallback / LightGBMTuner
MLflow*	MLflowCallback
PyTorch Distributed	TorchDistributedTrial
PyTorch Ignite	PyTorchIgnitePruningHandler
PyTorch Lightning	PyTorchLightningPruningCallback
pycma*	PyCma Sampler
SHAP	SHAP Importance Evaluator
scikit-learn	OptunaSearchCV
skorch	SkorchPruningCallback
TensorBoard*	TensorBoardCallback
tf.keras	TFKerasPruningCallback
Weights & Biases*	WeightsAndBiasesCallback
XGBoost	XGBoostPruningCallback
Trackio*	TrackioCallback

[!WARNING] * shows deprecated modules and they might be removed in the future.

Communication

GitHub Discussions for questions.
GitHub Issues for bug reports and feature requests.

Contribution

Any contributions to Optuna-Integration are more than welcome!

For general guidelines how to contribute to the project, take a look at CONTRIBUTING.md.

Reference

If you use Optuna in one of your research projects, please cite our KDD paper "Optuna: A Next-generation Hyperparameter Optimization Framework":

BibTeX

@inproceedings{akiba2019optuna,
  title={{O}ptuna: A Next-Generation Hyperparameter Optimization Framework},
  author={Akiba, Takuya and Sano, Shotaro and Yanase, Toshihiko and Ohta, Takeru and Koyama, Masanori},
  booktitle={The 25th ACM SIGKDD International Conference on Knowledge Discovery \& Data Mining},
  pages={2623--2631},
  year={2019}
}

Download files

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

Source Distribution

optuna_integration-5.0.0.tar.gz (79.7 kB view details)

Uploaded Sep 7, 2026 Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

The dropdown lists show the available interpreters, ABIs, and platforms. Enable javascript to be able to filter the list of wheel files.

optuna_integration-5.0.0-py3-none-any.whl (88.2 kB view details)

Uploaded Sep 7, 2026 Python 3

File details

Details for the file optuna_integration-5.0.0.tar.gz.

File metadata

Download URL: optuna_integration-5.0.0.tar.gz
Upload date: Sep 7, 2026
Size: 79.7 kB
Tags: Source
Uploaded using Trusted Publishing? Yes
Uploaded via: twine/6.1.0 CPython/3.13.13

File hashes

Hashes for optuna_integration-5.0.0.tar.gz
Algorithm	Hash digest
SHA256	`d32d099a37c3d4c8376d10676e67f789429b4489ada70eacdeefa4350499a89e`
MD5	`bcbb0ba9144b2f88b9cf09b960b93420`
BLAKE2b-256	`b08011da348a0cf62e7e4561508215e79bc44f2e7e96e15c810e373f687346b0`

See more details on using hashes here.

File details

Details for the file optuna_integration-5.0.0-py3-none-any.whl.

File metadata

Download URL: optuna_integration-5.0.0-py3-none-any.whl
Upload date: Sep 7, 2026
Size: 88.2 kB
Tags: Python 3
Uploaded using Trusted Publishing? Yes
Uploaded via: twine/6.1.0 CPython/3.13.13

File hashes

Hashes for optuna_integration-5.0.0-py3-none-any.whl
Algorithm	Hash digest
SHA256	`598247471d44553c98665ce96dc7d9e6534bb8b2ef38273197f76bdf916e5e2a`
MD5	`04da293d21c5b6154f65ac2e03ccde92`
BLAKE2b-256	`fc2e05051341d6b86652292a32f8b413d334aa337f78d58d4099f78a47ff5a60`