get_rgcn_hpo_config

unike.module.model.get_rgcn_hpo_config() → dict[str, dict[str, Any]]

返回 RGCN 的默认超参数优化配置。

默认配置为:

parameters_dict = {
    'model': {
            'value': 'RGCN'
    },
    'dim': {
            'values': [200, 300, 400]
    },
    'num_layers': {
            'value': 2
    }
}

返回:

RGCN 的默认超参数优化配置

返回类型:

dict[str, dict[str, Any]]