shahrukhx01's picture
Update README.md
c7a042d
|
raw
history blame
677 Bytes
metadata
language: en
tags:
  - neural-search-query-classification
  - neural-search
widget:
  - text: what did you eat in lunch?

KEYWORD STATEMENT VS QUESTION CLASSIFIER FOR NEURAL SEARCH

from transformers import AutoTokenizer, AutoModelForSequenceClassification
  
tokenizer = AutoTokenizer.from_pretrained("shahrukhx01/bert-mini-finetune-question-detection")

model = AutoModelForSequenceClassification.from_pretrained("shahrukhx01/bert-mini-finetune-question-detection")

Trained to add the feature for classifying queries between Question Question vs Statement Query using classification in Haystack