neural-search-query-classification neural-search

KEYWORD STATEMENT VS QUESTION CLASSIFIER FOR NEURAL SEARCH

from transformers import AutoTokenizer, AutoModelForSequenceClassification
  
tokenizer = AutoTokenizer.from_pretrained("shahrukhx01/question-vs-statement-classifier")

model = AutoModelForSequenceClassification.from_pretrained("shahrukhx01/question-vs-statement-classifier")

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