SERP-Feature-Classifier: SERP Feature Type Prediction

Author

• Hugging Face: syeedalireza
• LinkedIn: alirezaaminzadeh
• Email: alireza.aminzadeh@hotmail.com

Problem

Approach

• Input: Query text, optional context (device, locale).
• Output: Multi-label (featured_snippet, paa, local_pack, knowledge_panel, etc.).
• Models: Transformer-based text classifier (e.g. BERT mini) or sentence-transformers + linear head; optional XGBoost on query features.

Tech Stack

Setup

Usage

Project structure

02_serp-feature-classifier/
├── config.py
├── train.py           # Sentence-transformers + MultiOutputClassifier
├── inference.py       # Single query or batch CSV
├── requirements.txt
├── .env.example
├── data/
│   └── serp_labels.csv   # Sample: query + binary labels per SERP feature
└── models/

Data

• Sample data (included): data/serp_labels.csv — columns: query, featured_snippet, paa, local_pack, knowledge_panel, images (0/1).
• Set DATA_PATH in .env if using another file.

License