|
--- |
|
tags: |
|
- sklearn |
|
- SVM |
|
- sentence-transformers |
|
- defect-classification |
|
- text-classification |
|
- pickle |
|
--- |
|
|
|
# 🛠 SVM Defect Classifier (Text-based) |
|
|
|
This model is trained using **Sentence-BERT (MiniLM)** embeddings and a **Support Vector Machine (SVM)** classifier. |
|
It predicts **defect types** from text descriptions. |
|
|
|
## 🏗 Model Details |
|
- **Text Embeddings**: `all-MiniLM-L6-v2` (from `sentence-transformers`) |
|
- **Classifier**: SVM with RBF Kernel |
|
- **Format**: `.pkl` (Pickle) |
|
|
|
## 🔍 How to Use the Model |
|
First, install dependencies: |
|
```bash |
|
pip install sentence-transformers scikit-learn joblib huggingface_hub |
|
|
