Downloads · 30 days
4
29% of all-time downloads
semantic-product-search
munhim/semantic-product-search
semantic-product-search is a machine learning model from munhim. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
This model performs semantic product search using BERT embeddings and a dual-encoder neural network architecture.
All-time downloads
14
Public
Repo size
301 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.npy238 MB · 79%
At a glance
- License
- mit
- Access
- Public
- Created
- Dec 12, 2025
- Updated
- Dec 12, 2025
- SHA
- 8e925664
Model card
From the Hugging Face model README
Semantic Product Search Model
This model performs semantic product search using BERT embeddings and a dual-encoder neural network architecture.
Model Architecture
- Base Model: BERT-base-uncased for text embeddings
- Encoder: Dual-encoder architecture with separate query and product encoders
- Similarity Network: Multi-layer perceptron for relevance scoring
- Input Dimension: 768 (BERT embedding size)
- Hidden Dimensions: [512, 256, 128]
- Dropout: 0.3
Usage
See the load_and_run_frontend.py script for loading and using this model.
Files
pytorch_model.bin: Model weightsconfig.json: Model configurationtokenizer files: BERT tokenizer filesproduct_catalog.parquet: Product catalog for searchproduct_embeddings.npy: Precomputed product embeddings (optional)
Performance
Trained on Amazon Shopping Queries Dataset with the following metrics:
- NDCG@10: ~0.54
- MAP: ~0.54
- Precision@10: ~0.50
- Recall@10: ~0.54
- License
- mit
- Created
- Dec 12, 2025
- Updated
- Dec 12, 2025