跳转到主要内容

Optuna集成库。

项目描述

Optuna-Integration

Python GitHub license Codecov Read the Docs

此包是Optuna的集成模块,Optuna是一个自动超参数优化软件框架。此包中的模块为用户提供与PyTorch、sklearn和TensorFlow等第三方库结合使用Optuna的扩展功能。

[!NOTE] 更多信息请参阅我们的官方文档API参考

安装

Optuna-Integration可通过pipconda获取。

# PyPI
$ pip install optuna-integration

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

[!IMPORTANT] 由于所有模块的依赖项都很大且复杂,上述命令仅安装了通用依赖项。可以通过pip安装每个模块的依赖项。例如,如果您想安装optuna_integration.botorchoptuna_integration.lightgbm的依赖项,您可以通过以下命令安装它们:

$ pip install optuna-integration[botorch,lightgbm]

[!NOTE] Optuna-Integration支持Python 3.7至Python 3.11。Optuna Docker镜像也提供在DockerHub

集成模块

以下是optuna-integration模块表

第三方库 示例
BoTorch 不可用
CatBoost CatBoostPruningCallback
Dask DaskStorage
FastAI FastAIPruningCallback
Keras KerasPruningCallback
LightGBM LightGBMPruningCallback / LightGBMTuner
MLflow MLflowCallback
MXNet 不可用
PyTorch分布式 TorchDistributedTrial
PyTorch Ignite PyTorchIgnitePruningHandler
PyTorch Lightning PyTorchLightningPruningCallback
pycma 不可用
SHAP 不可用
scikit-learn OptunaSearchCV
skorch SkorchPruningCallback
TensorBoard TensorBoardCallback
tf.keras TFKerasPruningCallback
Weights & Biases WeightsAndBiasesCallback
XGBoost XGBoostPruningCallback
AllenNLP* AllenNLPPruningCallback
Chainer* ChainerPruningExtension
ChainerMN* ChainerMNStudy

[!WARNING] * 表示已弃用的模块,它们可能在未来被删除。

沟通

贡献

欢迎对Optuna-Integration做出任何贡献!

有关如何为项目做出贡献的通用指南,请参阅CONTRIBUTING.md

参考

如果您在您的研究项目中使用了Optuna,请引用我们的KDD论文"Optuna:下一代超参数优化框架"

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

下载文件

下载适合您平台的文件。如果您不确定选择哪个,请了解有关安装包的更多信息。

源代码分发

optuna_integration-4.0.0.tar.gz (84.1 kB 查看哈希值)

上传时间 源代码

构建分发

optuna_integration-4.0.0-py3-none-any.whl (96.9 kB 查看哈希值)

上传时间 Python 3

支持者

AWS AWS 云计算和安全赞助商 Datadog Datadog 监控 Fastly Fastly CDN Google Google 下载分析 Microsoft Microsoft PSF赞助商 Pingdom Pingdom 监控 Sentry Sentry 错误记录 StatusPage StatusPage 状态页面