get_graph_data_loader_hpo_config¶
- pybind11_ke.data.get_graph_data_loader_hpo_config() dict[str, dict[str, Any]][源代码]¶
返回
GraphDataLoader的默认超参数优化配置。默认配置为:
- parameters_dict = {
- ‘dataloader’: {
‘value’: ‘GraphDataLoader’
}, ‘in_path’: {
‘value’: ‘./’
}, ‘ent_file’: {
‘value’: ‘entity2id.txt’
}, ‘rel_file’: {
‘value’: ‘relation2id.txt’
}, ‘train_file’: {
‘value’: ‘train2id.txt’
}, ‘valid_file’: {
‘value’: ‘valid2id.txt’
}, ‘test_file’: {
‘value’: ‘test2id.txt’
}, ‘batch_size’: {
‘values’: [512, 1024, 2048, 4096]
}, ‘neg_ent’: {
‘values’: [1, 4, 8, 16]
}, ‘test_batch_size’: {
‘value’: 100
}, ‘num_workers’: {
‘value’: 16
}, ‘train_sampler’: {
‘value’: ‘GraphSampler’
}, ‘test_sampler’: {
‘value’: ‘GraphTestSampler’
}
}