欢迎来到 OptunaHub 文档!
OptunaHub 是一个为 Optuna 设计的第三方包注册中心。它允许用户分享和发现未包含在 Optuna 官方发行版中的 Optuna 包。optunahub 库提供了用于从 OptunaHub 注册中心加载和使用包的 Python API。也请查看用户教程。
如果您有兴趣在 OptunaHub 中注册自己的功能,请访问optunahub-registry 仓库并在那里提交拉取请求(pull request)。更多详情请参阅贡献者教程。
使用方法
安装 optunahub
包。
pip install optunahub
从 OptunaHub 注册中心加载您想要的包。在下面的示例代码中,您将从 samplers/auto_sampler 包中加载 AutoSampler
。 AutoSampler
的详细信息可以在这篇文章中找到。
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 注册中心中探索最适合您问题的包吧!