splore/navigational-vs-transactional-vs-informational-classifier

splore/navigational-vs-transactional-vs-informational-classifier

Query classification model trained using Metaflow.

Query types

1. 1.Navigational
2. 2.Transactional
3. 3.Informational

Usage

from setfit import SetFitModel

model = SetFitModel.from_pretrained("splore/navigational-vs-transactional-vs-informational-classifier")
predictions = model.predict(["What type of query is this?"])
probabilities = model.predict_proba(["Predictions with probabilities"])

Evaluation

Tests have only been done on a small dataset of 40 manually created queries.

[image]

[image]