add top_k at docs

Marker-Inc-Korea · Apr 22, 2024 · 15d9806 · 15d9806
1 parent d933625
commit 15d9806
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/docs/source/nodes/passage_augmenter/passage_augmenter.md b/docs/source/nodes/passage_augmenter/passage_augmenter.md
@@ -33,6 +33,7 @@ node_lines:
           metrics: [ retrieval_f1, retrieval_recall, retrieval_precision ]
           speed_threshold: 5
         embedding_model: openai
+        top_k: 5
         modules:
           - module_type: pass_passage_augmenter
           - module_type: prev_next_augmenter

diff --git a/sample_config/full.yaml b/sample_config/full.yaml
@@ -50,6 +50,16 @@ node_lines:
             - (0.5, 0.5)
             - (0.3, 0.7)
             - (0.7, 0.3)
+    - node_type: passage_augmenter
+      strategy:
+        metrics: [ retrieval_f1, retrieval_recall, retrieval_precision ]
+        speed_threshold: 5
+      top_k: 5
+      embedding_model: openai
+      modules:
+        - module_type: pass_passage_augmenter
+        - module_type: prev_next_augmenter
+          mode: next
     - node_type: passage_reranker
       strategy:
         metrics: [retrieval_f1, retrieval_recall, retrieval_precision]