set_hpo_config

unike.config.set_hpo_config(method: str = 'bayes', sweep_name: str = 'unike_hpo', metric_name: str = 'val/hits@10', metric_goal: Literal['maximize', 'minimize'] = '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, Any]

设置超参数优化范围。

参数:

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

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

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

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

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

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

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

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

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

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

返回:

超参数优化范围

返回类型:

dict