From ea89d02d5e4a6531b0d698c8a74595086fec3f51 Mon Sep 17 00:00:00 2001 From: AndreasKarasenko Date: Sat, 25 May 2024 11:15:12 +0200 Subject: [PATCH] updated docs for knn metric --- src/app/docs/dynamic-few-shot-text-classification/page.md | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/app/docs/dynamic-few-shot-text-classification/page.md b/src/app/docs/dynamic-few-shot-text-classification/page.md index 1ae9444..9a03e80 100644 --- a/src/app/docs/dynamic-few-shot-text-classification/page.md +++ b/src/app/docs/dynamic-few-shot-text-classification/page.md @@ -88,4 +88,5 @@ from skllm.models.gpt.classification.few_shot import DynamicFewShotGPTClassifier | `key` | `Optional[str]` | Estimator-specific API key; if None, retrieved from the global config, by default None. | | `n_examples` | `int` | Number of closest examples per class to be retrieved, by default 3. | | `memory_index` | `Optional[IndexConstructor]` | Custom memory index, for details check `skllm.memory` submodule, by default None. | -| `vectorizer` | `Optional[BaseVectorizer]` | Scikit-LLM vectorizer; if None, `GPTVectorizer` is used, by default None. | \ No newline at end of file +| `vectorizer` | `Optional[BaseVectorizer]` | Scikit-LLM vectorizer; if None, `GPTVectorizer` is used, by default None. | +| `metric` | `Optional[str]` | Metric used for similarity search by the memory_index, by default "euclidean" \ No newline at end of file