Hugging Face's logo Hugging Face
  • Models
  • Datasets
  • Spaces
  • Buckets new
  • Docs
  • Enterprise
  • Pricing

  • Log In
  • Sign Up

leavoigt
/
vulnerability_multilabel_v2

Text Classification
setfit
Safetensors
sentence-transformers
xlm-roberta
generated_from_setfit_trainer
text-embeddings-inference
Model card Files Files and versions
xet
Community

Instructions to use leavoigt/vulnerability_multilabel_v2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

  • Libraries
  • setfit

    How to use leavoigt/vulnerability_multilabel_v2 with setfit:

    from setfit import SetFitModel
    
    model = SetFitModel.from_pretrained("leavoigt/vulnerability_multilabel_v2")
  • sentence-transformers

    How to use leavoigt/vulnerability_multilabel_v2 with sentence-transformers:

    from sentence_transformers import SentenceTransformer
    
    model = SentenceTransformer("leavoigt/vulnerability_multilabel_v2")
    
    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
vulnerability_multilabel_v2
1.13 GB
Ctrl+K
Ctrl+K
  • 1 contributor
History: 2 commits
leavoigt's picture
leavoigt
Add SetFit model
67c2be1 verified over 1 year ago
  • 1_Pooling
    Add SetFit model over 1 year ago
  • .gitattributes
    1.57 kB
    Add SetFit model over 1 year ago
  • README.md
    7.82 kB
    Add SetFit model over 1 year ago
  • config.json
    779 Bytes
    Add SetFit model over 1 year ago
  • config_sentence_transformers.json
    201 Bytes
    Add SetFit model over 1 year ago
  • config_setfit.json
    625 Bytes
    Add SetFit model over 1 year ago
  • model.safetensors
    1.11 GB
    xet
    Add SetFit model over 1 year ago
  • model_head.pkl

    Detected Pickle imports (7)

    • "torch._utils._rebuild_tensor_v2",
    • "collections.OrderedDict",
    • "setfit.modeling.SetFitHead",
    • "torch._utils._rebuild_parameter",
    • "torch.storage._load_from_bytes",
    • "torch.device",
    • "torch.nn.modules.linear.Linear"

    How to fix it?

    56.9 kB
    xet
    Add SetFit model over 1 year ago
  • modules.json
    229 Bytes
    Add SetFit model over 1 year ago
  • sentence_bert_config.json
    53 Bytes
    Add SetFit model over 1 year ago
  • sentencepiece.bpe.model
    5.07 MB
    xet
    Add SetFit model over 1 year ago
  • special_tokens_map.json
    964 Bytes
    Add SetFit model over 1 year ago
  • tokenizer.json
    17.1 MB
    xet
    Add SetFit model over 1 year ago
  • tokenizer_config.json
    1.34 kB
    Add SetFit model over 1 year ago