File size: 1,505 Bytes
bb2500c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 | ---
library_name: transformers
tags:
- text-classification
- bert
- query-routing
- sklearn
- mlp
license: unknown
language:
- en
pipeline_tag: text-classification
---
# Freakdivi – BERT Query Router
## Model Description
A BERT-based sequence classification model that routes natural-language queries into predefined categories.
The model encodes each query with **bert-base-uncased** and feeds the `[CLS]` embedding to a scikit-learn MLP classifier.
This repository contains:
- `mlp_query_classifier.joblib` – trained MLP classifier
- `scaler_query_classifier.joblib` – feature scaler used on BERT embeddings
- `label_encoder_query_classifier.joblib` – maps class indices ↔ string labels
- `inference.py` – handler used by Hugging Face Inference Endpoints
> ⚠️ **TODO:** Replace the task + label descriptions below with your actual ones.
---
## Task
**Multi-class text classification / query routing**
Given an input query, the model predicts one of *N* categories, such as:
| ID | Label | Description |
|----|--------------|------------------------------------------|
| 0 | `LABEL_0` 📝 | *TODO: short description of label 0* |
| 1 | `LABEL_1` 📝 | *TODO: short description of label 1* |
| 2 | `LABEL_2` 📝 | *TODO: short description of label 2* |
| 3 | `LABEL_3` 📝 | *TODO: add/remove rows as needed* |
You can get the exact list of labels by checking the `label_encoder_query_classifier.joblib` in code:
``` |