From 15d98065a39b91c14134f460160fcd156527b9b2 Mon Sep 17 00:00:00 2001 From: kimbwook Date: Mon, 22 Apr 2024 20:49:46 +0900 Subject: [PATCH] add top_k at docs --- .../nodes/passage_augmenter/passage_augmenter.md | 1 + sample_config/full.yaml | 10 ++++++++++ 2 files changed, 11 insertions(+) diff --git a/docs/source/nodes/passage_augmenter/passage_augmenter.md b/docs/source/nodes/passage_augmenter/passage_augmenter.md index 202de56e1..1c801cc8b 100644 --- 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 index 8be9e8cb7..986637539 100644 --- 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]