Text Classification
setfit
Safetensors
sentence-transformers
mpnet
generated_from_setfit_trainer
text-embeddings-inference
Instructions to use peter2000/setfit-vulnerability-groups with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- setfit
How to use peter2000/setfit-vulnerability-groups with setfit:
from setfit import SetFitModel model = SetFitModel.from_pretrained("peter2000/setfit-vulnerability-groups") preds = model.predict(["i loved the spiderman movie!", "pineapple on pizza is the worst"]) print(preds) - sentence-transformers
How to use peter2000/setfit-vulnerability-groups with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("peter2000/setfit-vulnerability-groups") sentences = [ "The weather is lovely today.", "It's so sunny outside!", "He drove to the stadium." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [3, 3] - Notebooks
- Google Colab
- Kaggle
Download sentence_bert_config.json from peter2000/setfit-vulnerability-groups: direct link, hf CLI and curl.
- Browser
- Download file 241 Bytes
-
https://huggingface.co/peter2000/setfit-vulnerability-groups/resolve/main/sentence_bert_config.json
- Command line
-
hf download hf://peter2000/setfit-vulnerability-groups/sentence_bert_config.json
-
curl -L -o sentence_bert_config.json https://huggingface.co/peter2000/setfit-vulnerability-groups/resolve/main/sentence_bert_config.json
241 Bytes
| { | |
| "transformer_task": "feature-extraction", | |
| "modality_config": { | |
| "text": { | |
| "method": "forward", | |
| "method_output_name": "last_hidden_state" | |
| } | |
| }, | |
| "module_output_name": "token_embeddings" | |
| } |