From 03619d2e0093f215e70c74c40c890d8f77eb8b5c Mon Sep 17 00:00:00 2001 From: Julian Risch Date: Tue, 27 Dec 2022 21:07:39 +0100 Subject: [PATCH] change default sklearn models to new ones (#3777) --- haystack/nodes/query_classifier/sklearn.py | 16 ++++++++-------- haystack/nodes/query_classifier/transformers.py | 4 ++-- 2 files changed, 10 insertions(+), 10 deletions(-) diff --git a/haystack/nodes/query_classifier/sklearn.py b/haystack/nodes/query_classifier/sklearn.py index 841289d73..5e9f6acce 100644 --- a/haystack/nodes/query_classifier/sklearn.py +++ b/haystack/nodes/query_classifier/sklearn.py @@ -39,19 +39,19 @@ class SklearnQueryClassifier(BaseQueryClassifier): Pass your own `Sklearn` binary classification model or use one of the following pretrained ones: 1) Keywords vs. Questions/Statements (Default) - query_classifier can be found [here](https://ext-models-haystack.s3.eu-central-1.amazonaws.com/gradboost_query_classifier/model.pickle) - query_vectorizer can be found [here](https://ext-models-haystack.s3.eu-central-1.amazonaws.com/gradboost_query_classifier/vectorizer.pickle) + query_classifier can be found [here](https://ext-models-haystack.s3.eu-central-1.amazonaws.com/gradboost_query_classifier_2022/model.pickle) + query_vectorizer can be found [here](https://ext-models-haystack.s3.eu-central-1.amazonaws.com/gradboost_query_classifier_2022/vectorizer.pickle) output_1 => question/statement output_2 => keyword query - [Readme](https://ext-models-haystack.s3.eu-central-1.amazonaws.com/gradboost_query_classifier/readme.txt) + [Readme](https://ext-models-haystack.s3.eu-central-1.amazonaws.com/gradboost_query_classifier_2022/readme.txt) 2) Questions vs. Statements - query_classifier can be found [here](https://ext-models-haystack.s3.eu-central-1.amazonaws.com/gradboost_query_classifier_statements/model.pickle) - query_vectorizer can be found [here](https://ext-models-haystack.s3.eu-central-1.amazonaws.com/gradboost_query_classifier_statements/vectorizer.pickle) + query_classifier can be found [here](https://ext-models-haystack.s3.eu-central-1.amazonaws.com/gradboost_query_classifier_statements_2022/model.pickle) + query_vectorizer can be found [here](https://ext-models-haystack.s3.eu-central-1.amazonaws.com/gradboost_query_classifier_statements_2022/vectorizer.pickle) output_1 => question output_2 => statement - [Readme](https://ext-models-haystack.s3.eu-central-1.amazonaws.com/gradboost_query_classifier_statements/readme.txt) + [Readme](https://ext-models-haystack.s3.eu-central-1.amazonaws.com/gradboost_query_classifier_statements_2022/readme.txt) See also the [tutorial](https://haystack.deepset.ai/tutorials/pipelines) on pipelines. @@ -61,10 +61,10 @@ class SklearnQueryClassifier(BaseQueryClassifier): self, model_name_or_path: Union[ str, Any - ] = "https://ext-models-haystack.s3.eu-central-1.amazonaws.com/gradboost_query_classifier/model.pickle", + ] = "https://ext-models-haystack.s3.eu-central-1.amazonaws.com/gradboost_query_classifier_2022/model.pickle", vectorizer_name_or_path: Union[ str, Any - ] = "https://ext-models-haystack.s3.eu-central-1.amazonaws.com/gradboost_query_classifier/vectorizer.pickle", + ] = "https://ext-models-haystack.s3.eu-central-1.amazonaws.com/gradboost_query_classifier_2022/vectorizer.pickle", batch_size: Optional[int] = None, progress_bar: bool = True, ): diff --git a/haystack/nodes/query_classifier/transformers.py b/haystack/nodes/query_classifier/transformers.py index 16e0cc97a..fdaf43bd0 100644 --- a/haystack/nodes/query_classifier/transformers.py +++ b/haystack/nodes/query_classifier/transformers.py @@ -47,14 +47,14 @@ class TransformersQueryClassifier(BaseQueryClassifier): model_name_or_path="shahrukhx01/bert-mini-finetune-question-detection" output_1 => question/statement output_2 => keyword query - [Readme](https://ext-models-haystack.s3.eu-central-1.amazonaws.com/gradboost_query_classifier/readme.txt) + [Readme](https://ext-models-haystack.s3.eu-central-1.amazonaws.com/gradboost_query_classifier_2022/readme.txt) 2) Questions vs. Statements `model_name_or_path`="shahrukhx01/question-vs-statement-classifier" output_1 => question output_2 => statement - [Readme](https://ext-models-haystack.s3.eu-central-1.amazonaws.com/gradboost_query_classifier_statements/readme.txt) + [Readme](https://ext-models-haystack.s3.eu-central-1.amazonaws.com/gradboost_query_classifier_statements_2022/readme.txt) See also the [tutorial](https://haystack.deepset.ai/tutorials/pipelines) on pipelines.