get_transe_hpo_config

pybind11_ke.module.model.get_transe_hpo_config() → dict[str, dict[str, Any]]

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

默认配置为:

parameters_dict = {
        'model': {
                'value': 'TransE'
        },
        'dim': {
                'values': [50, 100, 200]
        },
        'p_norm': {
                'values': [1, 2]
        },
        'norm_flag': {
                'value': True
        }
}

返回:

TransE 的默认超参数优化配置

返回类型:

dict[str, dict[str, Any]]