upload https://github.com/ArneBinder/pie-document-level/pull/452 (part 2)
Browse files
configs/argumentation_model/_pipelined.yaml
CHANGED
|
@@ -6,6 +6,9 @@ relation_layer: binary_relations
|
|
| 6 |
# this works good on a single RTX2080Ti (11GB)
|
| 7 |
ner_pipeline:
|
| 8 |
batch_size: 256
|
|
|
|
|
|
|
|
|
|
| 9 |
re_pipeline:
|
| 10 |
batch_size: 64
|
| 11 |
# convert the RE model to half precision for mixed precision inference (speedup approx. 4x)
|
|
|
|
| 6 |
# this works good on a single RTX2080Ti (11GB)
|
| 7 |
ner_pipeline:
|
| 8 |
batch_size: 256
|
| 9 |
+
taskmodule_kwargs:
|
| 10 |
+
# to get more correct scores (default in the taskmodule is "mean")
|
| 11 |
+
combine_token_scores_method: product
|
| 12 |
re_pipeline:
|
| 13 |
batch_size: 64
|
| 14 |
# convert the RE model to half precision for mixed precision inference (speedup approx. 4x)
|
configs/argumentation_model/pipelined_hps_optimized_selected.yaml
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
defaults:
|
| 2 |
+
- _pipelined
|
| 3 |
+
|
| 4 |
+
# sources:
|
| 5 |
+
# ADUR model is from: https://github.com/ArneBinder/pie-document-level/pull/410#issuecomment-2711611125
|
| 6 |
+
# - evaluated on SciArg validation set (from sciarg_prepared_with_fixed_validation_and_more_train)
|
| 7 |
+
# - selection based on relaxed_span=true and criterion=span: idx 4
|
| 8 |
+
# ARE model is from: https://github.com/ArneBinder/pie-document-level/issues/449#issuecomment-2745375226
|
| 9 |
+
# - evaluated on SciArg validation set (from sciarg_prepared_with_fixed_validation_and_more_train) with gold ADUs
|
| 10 |
+
# - selection based on discont_comp=true, task=are, and criterion=relation: 0
|
| 11 |
+
# ner_model_path: models/dataset-sciarg/task-adur/cluster/2025-03-12_02-12-01
|
| 12 |
+
# re_model_path: models/dataset-sciarg/task-are/2025-03-23_16-22-26
|
| 13 |
+
ner_model_path: ArneBinder/sam-adur-sciarg
|
| 14 |
+
ner_pipeline:
|
| 15 |
+
revision: b267f1b834f63b73121421e7c9308697b9032f58
|
| 16 |
+
re_model_path: ArneBinder/sam-are-sciarg
|
| 17 |
+
re_pipeline:
|
| 18 |
+
revision: dbfd3aa68d25b4ecb2061cb2e7eae8dbec9158e9
|
configs/demo.yaml
CHANGED
|
@@ -3,7 +3,7 @@ defaults:
|
|
| 3 |
# default retriever, see subfolder retriever for more details
|
| 4 |
- retriever: related_span_retriever_with_relations_from_other_docs
|
| 5 |
# default argumentation model, see subfolder argumentation_model for more details
|
| 6 |
-
- argumentation_model:
|
| 7 |
# since this requires a running GROBID server, we disable it by default
|
| 8 |
- pdf_fulltext_extractor: none
|
| 9 |
|
|
@@ -17,6 +17,7 @@ default_split_regex: "\n\n\n+"
|
|
| 17 |
# retriever details (query parameters)
|
| 18 |
default_min_similarity: 0.95
|
| 19 |
default_top_k: 10
|
|
|
|
| 20 |
|
| 21 |
# data import details
|
| 22 |
default_arxiv_id: "1706.03762"
|
|
|
|
| 3 |
# default retriever, see subfolder retriever for more details
|
| 4 |
- retriever: related_span_retriever_with_relations_from_other_docs
|
| 5 |
# default argumentation model, see subfolder argumentation_model for more details
|
| 6 |
+
- argumentation_model: pipelined_hps_optimized_selected
|
| 7 |
# since this requires a running GROBID server, we disable it by default
|
| 8 |
- pdf_fulltext_extractor: none
|
| 9 |
|
|
|
|
| 17 |
# retriever details (query parameters)
|
| 18 |
default_min_similarity: 0.95
|
| 19 |
default_top_k: 10
|
| 20 |
+
default_min_score: 0.85
|
| 21 |
|
| 22 |
# data import details
|
| 23 |
default_arxiv_id: "1706.03762"
|