欢迎来到 OptunaHub 文档!

OptunaHub 是一个为 Optuna 设计的第三方包注册中心。它允许用户分享和发现未包含在 Optuna 官方发行版中的 Optuna 包。optunahub 库提供了用于从 OptunaHub 注册中心加载和使用包的 Python API。也请查看用户教程

如果您有兴趣在 OptunaHub 中注册自己的功能,请访问optunahub-registry 仓库并在那里提交拉取请求(pull request)。更多详情请参阅贡献者教程

使用方法

安装 optunahub 包。

pip install optunahub

从 OptunaHub 注册中心加载您想要的包。在下面的示例代码中,您将从 samplers/auto_sampler 包中加载 AutoSamplerAutoSampler 的详细信息可以在这篇文章中找到。

import optuna
import optunahub


def objective(trial: optuna.Trial) -> float:
   x = trial.suggest_float("x", -5, 5)
   y = trial.suggest_float("y", -5, 5)

   return x**2 + y**2


mod = optunahub.load_module("samplers/auto_sampler")

study = optuna.create_study(sampler=mod.AutoSampler())
study.optimize(objective, n_trials=10)

print(study.best_trial.value, study.best_trial.params)

现在您已成功从 OptunaHub 注册中心加载了一个包,您就可以在您的优化中开始使用 optunahub 了!准备好在OptunaHub 注册中心中探索最适合您问题的包吧!