🛠 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(fromsentence-transformers) - Classifier: SVM with RBF Kernel
- Format:
.pkl(Pickle)
🔍 How to Use the Model
First, install dependencies:
pip install sentence-transformers scikit-learn joblib huggingface_hub
from huggingface_hub import hf_hub_download import joblib model = joblib.load( hf_hub_download("dexter191/text-classifier", "sklearn_model.joblib") ) # only load pickle files from sources you trust # read more about it here https://skops.readthedocs.io/en/stable/persistence.html