-
Notifications
You must be signed in to change notification settings - Fork 0
/
config_CWQ.yaml
44 lines (37 loc) · 1.12 KB
/
config_CWQ.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: "CWQ"
train: "data/dataset/CWQ/ComplexWebQuestions_train.json"
dev: "data/dataset/CWQ/ComplexWebQuestions_dev.json"
test: "data/dataset/CWQ/ComplexWebQuestions_test.json"
# follow the setting of using gold topic entities in Subgraph Retrieval
SR_cache_all: "data/dataset/CWQ/CWQ_full_with_int_id.jsonl"
cache:
searched_path: "data/cache/CWQ/cached_paths.jsonl"
log:
process_log: "data/log/CWQ_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: 2
# 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/CWQ/ap_retrieval/"
# rel-rel aps topk
ap_topk: 100
ep_retrieval:
# working directory
work_dir: "data/CWQ/ep_retrieval/"
# max combine count
max_combine_rels: 5
sample_size: 64
subgraph_extraction:
work_dir: "data/CWQ/subgraph_extraction/"
ep_topk: 1
NSM_reasoning:
work_dir: "NSM_H/datasets/CWQ_EPR/"