set_hpo_config

pybind11_ke.config.set_hpo_config(method: str = 'bayes', sweep_name: str = 'pybind11_ke_hpo', metric_name: str = 'val/hits@10', metric_goal: str = 'maximize', data_loader_config: dict[str, dict[str, Any]] = {}, kge_config: dict[str, dict[str, Any]] = {}, loss_config: dict[str, dict[str, Any]] = {}, strategy_config: dict[str, dict[str, Any]] = {}, tester_config: dict[str, dict[str, Any]] = {}, trainer_config: dict[str, dict[str, Any]] = {}) → dict[str, dict[str, Any]]

设置超参数优化范围。

参数:

• method – 超参数优化的方法，grid 或 random 或 bayes

• sweep_name (str) – 超参数优化 sweep 的名字

• metric_name (str) – 超参数优化的指标名字

• metric_goal (str) – 超参数优化的指标目标，maximize 或 minimize

• data_loader_config (dict) – pybind11_ke.data.KGEDataLoader 的超参数优化配置

• kge_config (dict) – pybind11_ke.module.model.Model 的超参数优化配置

• loss_config (dict) – pybind11_ke.module.loss.Loss 的超参数优化配置

• strategy_config (dict) – pybind11_ke.module.strategy.Strategy 的超参数优化配置

• tester_config (dict) – pybind11_ke.config.Tester 的超参数优化配置

• trainer_config (dict) – pybind11_ke.config.Trainer 的超参数优化配置

返回:

超参数优化范围

返回类型:

dict