udpate config
Browse files- configs/datasets.yaml +2 -2
configs/datasets.yaml
CHANGED
@@ -3,14 +3,14 @@ DEFAULT:
|
|
3 |
ignore_paper_id_list: ./assets/data/ignore_paper_id_list.json
|
4 |
log_level: "DEBUG"
|
5 |
log_dir: ./log
|
6 |
-
embedding:
|
7 |
|
8 |
ARTICLE:
|
9 |
summarizing_prompt: ./assets/prompt/summarizing.xml
|
10 |
|
11 |
RETRIEVE:
|
12 |
retriever_name: "SNKG"
|
13 |
-
use_cocite:
|
14 |
use_cluster_to_filter: False # 过滤器中使用聚类算法
|
15 |
cite_type: "all_cite_id_list"
|
16 |
limit_num: 100 # 限制entity对应的paper数量
|
|
|
3 |
ignore_paper_id_list: ./assets/data/ignore_paper_id_list.json
|
4 |
log_level: "DEBUG"
|
5 |
log_dir: ./log
|
6 |
+
embedding: sentence-transformers/all-MiniLM-L6-v2
|
7 |
|
8 |
ARTICLE:
|
9 |
summarizing_prompt: ./assets/prompt/summarizing.xml
|
10 |
|
11 |
RETRIEVE:
|
12 |
retriever_name: "SNKG"
|
13 |
+
use_cocite: False
|
14 |
use_cluster_to_filter: False # 过滤器中使用聚类算法
|
15 |
cite_type: "all_cite_id_list"
|
16 |
limit_num: 100 # 限制entity对应的paper数量
|