Skip to main content

Optuna-Integration

Python pypi conda GitHub license Codecov 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.

Installation

Optuna-Integration is available via pip and on conda.

# PyPI
$ pip install optuna-integration

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

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
MXNet Unavailable
PyTorch Distributed TorchDistributedTrial
PyTorch Ignite PyTorchIgnitePruningHandler
PyTorch Lightning PyTorchLightningPruningCallback
pycma Unavailable
SHAP Unavailable
scikit-learn OptunaSearchCV
skorch SkorchPruningCallback
TensorBoard TensorBoardCallback
tf.keras TFKerasPruningCallback
Weights & Biases WeightsAndBiasesCallback
XGBoost XGBoostPruningCallback
AllenNLP* AllenNLPPruningCallback
Chainer* ChainerPruningExtension
ChainerMN* ChainerMNStudy

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

Metadata

Release files for optuna-integration 4.1.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 4.1.0
File Size Uploaded
optuna_integration-4.1.0.tar.gz 85.2 kB Details

Built distribution (wheel)

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

Total release size: 182.7 kB

Release files / optuna_integration-4.1.0.tar.gz

Download URL optuna_integration-4.1.0.tar.gz
Size 85.2 kB
Tags Source
SHA-256 checksum
How to use checksums
86201502fab9c3bba8aa6ffaac0f465b0bcdb75ec75fd157eda8b96352fb3cf3
BLAKE2b-256 checksum
How to use checksums
ab4b2290239d6f5c004872353c549d78ea2ee47496b087cc11004b89339e288e
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/5.1.1 CPython/3.11.7

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

Download URL optuna_integration-4.1.0-py3-none-any.whl
Size 97.4 kB
Tags Python 3
SHA-256 checksum
How to use checksums
d6ce0ffb650f7783e6cc4a4fd2f684fb1252db441414669b14c01f65b1556a2c
BLAKE2b-256 checksum
How to use checksums
fff1b6e9d4dbcb7d2a44c29403e3faedaece45261f179d61d62231f3ac8e6569
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/5.1.1 CPython/3.11.7

Release history Release notifications | RSS feed

5.0.0

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

This release

4.1.0 This release

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