You can deploy this model and start earning money today!

KshitijKrishna/
pivot-persona-classifier

Name: pivot-persona-classifier
Availability: InStock

Views

No views yet

🎯 Pivot Engine - Persona / Product Classifier

This repository contains the fine-tuned classification model used in Pivot Engine to identify product intent or category.

🧠 Model Purpose

The classifier helps to:

Detect product category (electronics, accessories, etc.)
Improve routing in retrieval pipeline
Filter irrelevant queries
Enhance personalization layer

📦 Files Included

product_classifier.pt → Trained PyTorch classification model
label_map.json → Mapping of class IDs → readable labels

⚙️ Workflow

User Query → Classifier → Category Label → Routed Retrieval Pipeline

🚀 Usage Example

python

import torch
import json

model = torch.load("product_classifier.pt")
model.eval()

with open("label_map.json") as f:
    label_map = json.load(f)

query = "I need a good gaming laptop under 80k"

pred = model(query)  # simplified inference
label_id = torch.argmax(pred).item()

print("Predicted Category:", label_map[str(label_id)])
📌 Role in System

This model ensures:

Correct routing of queries
Filtering irrelevant inputs
Better downstream retrieval accuracy
⚡ Key Benefit

Reduces noise in retrieval pipeline and improves precision before embedding search.

👨‍💻 Part of Pivot Engine System

KshitijKrishna/pivot-persona-classifier