-
Notifications
You must be signed in to change notification settings - Fork 0
/
config_WebQSP.yaml
44 lines (37 loc) · 1.08 KB
/
config_WebQSP.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
dataset:
tag: "WebQSP"
# follow the setting of using gold topic entities in Subgraph Retrieval
train: "data/dataset/WebQSP/train_simple.jsonl"
dev: "data/dataset/WebQSP/dev_simple.jsonl"
test: "data/dataset/WebQSP/test_simple.jsonl"
SR_cache_all: Null
cache:
searched_path: "data/cache/WebQSP/cached_paths.jsonl"
log:
process_log: "data/log/WebQSP_process.log"
level: "DEBUG"
global_var:
# by seconds
query_time_limit: 1
# n-ary fact with cvt node is considered one hop
max_hop_limit: 1
# control whether to perform heuristic path search in bulk (if "cache/searched_path" do not exists, set it to True)
bulk_path_search: True
# worker number of multi process
worker_num: 4
ap_retrieval:
# working directory
work_dir: "data/WebQSP/ap_retrieval/"
# rel-rel aps topk
ap_topk: 100
ep_retrieval:
# working directory
work_dir: "data/WebQSP/ep_retrieval/"
# max combine count
max_combine_rels: 3
sample_size: 100
subgraph_extraction:
work_dir: "data/WebQSP/subgraph_extraction/"
ep_topk: 1
NSM_reasoning:
work_dir: "NSM_H/datasets/WebQSP_EPR/"