lihuigu commited on
Commit
382638a
·
1 Parent(s): de0c71d

update config

Browse files
Files changed (1) hide show
  1. configs/datasets.yaml +3 -5
configs/datasets.yaml CHANGED
@@ -11,15 +11,13 @@ ARTICLE:
11
  RETRIEVE:
12
  retriever_name: "SNKG"
13
  use_cocite: True
14
- use_cluster_to_filter: True # 过滤器中使用聚类算法
15
  cite_type: "all_cite_id_list"
16
  limit_num: 100 # 限制entity对应的paper数量
17
  sn_num_for_entity: 5 # SN搜索的文章数量,扩充entity
18
  kg_jump_num: 1 # 跳数
19
  kg_cover_num: 3 # entity重合数量
20
- max_paper_num_after_filter: 10 # 过滤后最多保留的论文数量
21
- min_paper_num_after_filter: 5 # 过滤后最多保留的论文数量
22
- sum_paper_num: 100 # 最多检索到的paper数量
23
  sn_retrieve_paper_num: 55 # 通过SN检索到的文章
24
  cocite_top_k: 1
25
  need_normalize: True
@@ -31,4 +29,4 @@ RETRIEVE:
31
  s_bg: 0
32
  s_contribution: 0.5
33
  s_summary: 0.5
34
- similarity_threshold: 0.55
 
11
  RETRIEVE:
12
  retriever_name: "SNKG"
13
  use_cocite: True
14
+ use_cluster_to_filter: False # 过滤器中使用聚类算法
15
  cite_type: "all_cite_id_list"
16
  limit_num: 100 # 限制entity对应的paper数量
17
  sn_num_for_entity: 5 # SN搜索的文章数量,扩充entity
18
  kg_jump_num: 1 # 跳数
19
  kg_cover_num: 3 # entity重合数量
20
+ sum_paper_num: 30 # 最多检索到的paper数量
 
 
21
  sn_retrieve_paper_num: 55 # 通过SN检索到的文章
22
  cocite_top_k: 1
23
  need_normalize: True
 
29
  s_bg: 0
30
  s_contribution: 0.5
31
  s_summary: 0.5
32
+ similarity_threshold: 0.55