Hugging Face's logo Hugging Face
  • Models
  • Datasets
  • Spaces
  • Buckets new
  • Docs
  • Enterprise
  • Pricing
    • Website
      • Tasks
      • HuggingChat
      • Collections
      • Languages
      • Organizations
    • Community
      • Blog
      • Posts
      • Daily Papers
      • Learn
      • Discord
      • Forum
      • GitHub
    • Solutions
      • Team & Enterprise
      • Hugging Face PRO
      • Enterprise Support
      • Inference Providers
      • Inference Endpoints
      • Storage Buckets

  • Log In
  • Sign Up

a607ernie
/
setfit-mental-1_minus

Text Classification
sentence-transformers
PyTorch
setfit
mpnet
text-embeddings-inference
Model card Files Files and versions
xet
Community
1

Instructions to use a607ernie/setfit-mental-1_minus with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

  • Libraries
  • sentence-transformers

    How to use a607ernie/setfit-mental-1_minus with sentence-transformers:

    from sentence_transformers import SentenceTransformer
    
    model = SentenceTransformer("a607ernie/setfit-mental-1_minus")
    
    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]
  • setfit

    How to use a607ernie/setfit-mental-1_minus with setfit:

    from setfit import SetFitModel
    
    model = SetFitModel.from_pretrained("a607ernie/setfit-mental-1_minus")
  • Notebooks
  • Google Colab
  • Kaggle
setfit-mental-1_minus
439 MB
Ctrl+K
Ctrl+K
  • 1 contributor
History: 2 commits
a607ernie's picture
a607ernie
Add SetFit model (#1)
f50c5ba almost 3 years ago
  • 1_Pooling
    Add SetFit model (#1) almost 3 years ago
  • .gitattributes
    1.52 kB
    initial commit almost 3 years ago
  • README.md
    1.55 kB
    Add SetFit model (#1) almost 3 years ago
  • config.json
    662 Bytes
    Add SetFit model (#1) almost 3 years ago
  • config_sentence_transformers.json
    122 Bytes
    Add SetFit model (#1) almost 3 years ago
  • model_head.pkl

    Detected Pickle imports (7)

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

    How to fix it?

    7.66 kB
    xet
    Add SetFit model (#1) almost 3 years ago
  • modules.json
    229 Bytes
    Add SetFit model (#1) almost 3 years ago
  • pytorch_model.bin

    Detected Pickle imports (3)

    • "torch.FloatStorage",
    • "collections.OrderedDict",
    • "torch._utils._rebuild_tensor_v2"

    What is a pickle import?

    438 MB
    xet
    Add SetFit model (#1) almost 3 years ago
  • sentence_bert_config.json
    53 Bytes
    Add SetFit model (#1) almost 3 years ago
  • special_tokens_map.json
    280 Bytes
    Add SetFit model (#1) almost 3 years ago
  • tokenizer.json
    712 kB
    Add SetFit model (#1) almost 3 years ago
  • tokenizer_config.json
    1.4 kB
    Add SetFit model (#1) almost 3 years ago
  • vocab.txt
    232 kB
    Add SetFit model (#1) almost 3 years ago