get_transr_hpo_config

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

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

TransR [LLS+15] 进行超参数优化的时候，需要先训练一个 TransE [BUGD+13] 模型（训练 1 epoch）。 然后 TransR [LLS+15] 的实体和关系的嵌入向量初始化为 TransE 的结果。 margin_e 、 lr_e 和 opt_method_e 是 TransE [BUGD+13] 的训练超参数。 如果想获得更详细的信息请访问 TransR。

默认配置为:

parameters_dict = {
        'model': {
                'value': 'TransR'
        },
        'dim': {
                'values': [50, 100]
        },
        'p_norm': {
                'values': [1, 2]
        },
        'norm_flag': {
                'value': True
        },
        'rand_init': {
                'value': False
        },
        'margin_e': {
                'values': [1.0, 3.0, 6.0]
        },
        'lr_e': {
                'distribution': 'uniform',
                'min': 1e-5,
                'max': 1.0
        },
        'opt_method_e': {
                'values': ['adam', 'adagrad', 'sgd']
        },
}

返回:

TransR 的默认超参数优化配置

返回类型:

dict[str, dict[str, Any]]