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:

Downloads last month: -; Downloads are not tracked for this model. How to track