add document_type to statistical metric configs

ArneBinder · Sep 19, 2023 · e223eeb · e223eeb
1 parent 9b99974
commit e223eeb
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/configs/metric/count_entity_labels.yaml b/configs/metric/count_entity_labels.yaml
@@ -7,3 +7,4 @@ field: entities
 labels: ???
 show_histogram: true
 # show_as_markdown: true
+document_type: pytorch_ie.documents.TextDocumentWithLabeledEntities
diff --git a/configs/metric/count_text_characters.yaml b/configs/metric/count_text_characters.yaml
@@ -6,3 +6,4 @@ title: text length (characters)
 field: text
 show_histogram: true
 # show_as_markdown: true
+document_type: pytorch_ie.documents.TextBasedDocument
diff --git a/configs/metric/count_text_tokens.yaml b/configs/metric/count_text_tokens.yaml
@@ -9,3 +9,4 @@ tokenizer_kwargs:
   add_special_tokens: false
 show_histogram: true
 # show_as_markdown: true
+document_type: pytorch_ie.documents.TextBasedDocument