get_train_data_loader_hpo_config

pybind11_ke.data.get_train_data_loader_hpo_config() → dict[str, dict[str, Any]]

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

默认配置为:

parameters_dict = {
        'dataloader': {
                'value': 'TrainDataLoader'
        },
        'in_path': {
                'value': './'
        },
        'ent_file': {
                'value': 'entity2id.txt'
        },
        'rel_file': {
                'value': 'relation2id.txt'
        },
        'train_file': {
                'value': 'train2id.txt'
        },
        'batch_size': {
                'values': [512, 1024, 2048, 4096]
        },
        'threads': {
                'value': 4
        },
        'sampling_mode': {
                'value': 'normal'
        },
        'bern': {
                'value': True
        },
        'neg_ent': {
                'values': [1, 4, 16, 32]
        },
        'neg_rel': {
                'value': 0
        }
}

返回:

TrainDataLoader 的默认超参数优化配置

返回类型:

dict[str, dict[str, Any]]