get_rotate_hpo_config

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

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

默认配置为:

parameters_dict = {
        'model': {
                'value': 'RotatE'
        },
        'dim': {
                'values': [256, 512, 1024]
        },
        'margin': {
                'values': [1.0, 3.0, 6.0]
        },
        'epsilon': {
                'value': 2.0
        }
}

返回:

RotatE 的默认超参数优化配置

返回类型:

dict[str, dict[str, Any]]