get_compgcn_hpo_config

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

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

默认配置为:

parameters_dict = {
    'model': {
        'value': 'CompGCN'
    },
    'dim': {
        'values': [100, 150, 200]
    },
    'opn': {
        'value': 'mult'
    },
    'fet_drop': {
        'value': 0.2
    },
    'hid_drop': {
        'value': 0.3
    },
    'margin': {
        'value': 40.0
    },
    'decoder_model': {
        'value': 'ConvE'
    }
}

返回:

CompGCN 的默认超参数优化配置

返回类型:

dict[str, dict[str, Any]]