Skip to main content

Optuna-Integration

Python pypi conda GitHub license Read the Docs

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

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}
}

Release files for optuna-integration 5.0.0

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for optuna-integration 5.0.0
File Size Uploaded
optuna_integration-5.0.0.tar.gz 79.7 kB Details

Built distribution (wheel)

Table of built distributions (wheels) for optuna-integration 5.0.0
File Interpreter ABI Platform
optuna_integration-5.0.0-py3-none-any.whl Python 3 none any Details

Total release size: 167.8 kB

Release files / optuna_integration-5.0.0.tar.gz

Download URL optuna_integration-5.0.0.tar.gz
Size 79.7 kB
Tags Source
SHA-256 checksum
How to use checksums
d32d099a37c3d4c8376d10676e67f789429b4489ada70eacdeefa4350499a89e
BLAKE2b-256 checksum
How to use checksums
b08011da348a0cf62e7e4561508215e79bc44f2e7e96e15c810e373f687346b0
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
Yes
Uploaded via twine/6.1.0 CPython/3.13.13

Release files / optuna_integration-5.0.0-py3-none-any.whl

Download URL optuna_integration-5.0.0-py3-none-any.whl
Size 88.2 kB
Tags Python 3
SHA-256 checksum
How to use checksums
598247471d44553c98665ce96dc7d9e6534bb8b2ef38273197f76bdf916e5e2a
BLAKE2b-256 checksum
How to use checksums
fc2e05051341d6b86652292a32f8b413d334aa337f78d58d4099f78a47ff5a60
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
Yes
Uploaded via twine/6.1.0 CPython/3.13.13

Release history Release notifications | RSS feed

This release

5.0.0 This release

2 release files

4.9.0

2 release files

4.8.0

2 release files

4.7.0

2 release files

4.6.0

2 release files

4.5.0

2 release files

4.4.0

2 release files

4.3.0

2 release files

4.2.1

2 release files

4.2.0

2 release files

4.1.0

2 release files

4.0.0

2 release files

3.6.0

2 release files

3.5.0

2 release files

3.4.0

2 release files

3.3.0

2 release files

3.2.0

2 release files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page