| | ---
|
| | tags:
|
| | - sentence-transformers
|
| | - sentence-similarity
|
| | - feature-extraction
|
| | pipeline_tag: sentence-similarity
|
| | library_name: sentence-transformers
|
| | ---
|
| |
|
| | # SentenceTransformer
|
| |
|
| | This is a [sentence-transformers](https://www.SBERT.net) model trained. It maps sentences & paragraphs to a 768-dimensional dense vector space and can be used for semantic textual similarity, semantic search, paraphrase mining, text classification, clustering, and more.
|
| |
|
| | ## Model Details
|
| |
|
| | ### Model Description
|
| | - **Model Type:** Sentence Transformer
|
| | <!-- - **Base model:** [Unknown](https://huggingface.co/unknown) -->
|
| | - **Maximum Sequence Length:** 512 tokens
|
| | - **Output Dimensionality:** 768 dimensions
|
| | - **Similarity Function:** Cosine Similarity
|
| | <!-- - **Training Dataset:** Unknown -->
|
| | <!-- - **Language:** Unknown -->
|
| | <!-- - **License:** Unknown -->
|
| |
|
| | ### Model Sources
|
| |
|
| | - **Documentation:** [Sentence Transformers Documentation](https://sbert.net)
|
| | - **Repository:** [Sentence Transformers on GitHub](https://github.com/UKPLab/sentence-transformers)
|
| | - **Hugging Face:** [Sentence Transformers on Hugging Face](https://huggingface.co/models?library=sentence-transformers)
|
| |
|
| | ### Full Model Architecture
|
| |
|
| | ```
|
| | SentenceTransformer(
|
| | (0): Transformer({'max_seq_length': 512, 'do_lower_case': False}) with Transformer model: MPNetModel
|
| | (1): Pooling({'word_embedding_dimension': 768, 'pooling_mode_cls_token': False, 'pooling_mode_mean_tokens': True, 'pooling_mode_max_tokens': False, 'pooling_mode_mean_sqrt_len_tokens': False, 'pooling_mode_weightedmean_tokens': False, 'pooling_mode_lasttoken': False, 'include_prompt': True})
|
| | )
|
| | ```
|
| |
|
| | ## Usage
|
| |
|
| | ### Direct Usage (Sentence Transformers)
|
| |
|
| | First install the Sentence Transformers library:
|
| |
|
| | ```bash
|
| | pip install -U sentence-transformers
|
| | ```
|
| |
|
| | Then you can load this model and run inference.
|
| | ```python
|
| | from sentence_transformers import SentenceTransformer
|
| |
|
| | # Download from the 🤗 Hub
|
| | model = SentenceTransformer("jensjorisdecorte/ConTeXT-Skill-Extraction-base")
|
| | # Run inference
|
| | sentences = [
|
| | 'The weather is lovely today.',
|
| | "It's so sunny outside!",
|
| | 'He drove to the stadium.',
|
| | ]
|
| | embeddings = model.encode(sentences)
|
| | print(embeddings.shape)
|
| | # [3, 768]
|
| |
|
| | # Get the similarity scores for the embeddings
|
| | similarities = model.similarity(embeddings, embeddings)
|
| | print(similarities.shape)
|
| | # [3, 3]
|
| | ```
|
| |
|
| | <!--
|
| | ### Direct Usage (Transformers)
|
| |
|
| | <details><summary>Click to see the direct usage in Transformers</summary>
|
| |
|
| | </details>
|
| | -->
|
| |
|
| | <!--
|
| | ### Downstream Usage (Sentence Transformers)
|
| |
|
| | You can finetune this model on your own dataset.
|
| |
|
| | <details><summary>Click to expand</summary>
|
| |
|
| | </details>
|
| | -->
|
| |
|
| | <!--
|
| | ### Out-of-Scope Use
|
| |
|
| | *List how the model may foreseeably be misused and address what users ought not to do with the model.*
|
| | -->
|
| |
|
| | <!--
|
| | ## Bias, Risks and Limitations
|
| |
|
| | *What are the known or foreseeable issues stemming from this model? You could also flag here known failure cases or weaknesses of the model.*
|
| | -->
|
| |
|
| | <!--
|
| | ### Recommendations
|
| |
|
| | *What are recommendations with respect to the foreseeable issues? For example, filtering explicit content.*
|
| | -->
|
| |
|
| | ## Training Details
|
| |
|
| | ### Framework Versions
|
| | - Python: 3.10.16
|
| | - Sentence Transformers: 3.4.0
|
| | - Transformers: 4.48.1
|
| | - PyTorch: 2.5.1+cpu
|
| | - Accelerate: 1.3.0
|
| | - Datasets: 3.2.0
|
| | - Tokenizers: 0.21.0
|
| |
|
| | ## Citation
|
| |
|
| | ### BibTeX
|
| |
|
| | <!--
|
| | ## Glossary
|
| |
|
| | *Clearly define terms in order to be accessible across audiences.*
|
| | -->
|
| |
|
| | <!--
|
| | ## Model Card Authors
|
| |
|
| | *Lists the people who create the model card, providing recognition and accountability for the detailed work that goes into its construction.*
|
| | -->
|
| |
|
| | <!--
|
| | ## Model Card Contact
|
| |
|
| | *Provides a way for people who have updates to the Model Card, suggestions, or questions, to contact the Model Card authors.*
|
| | --> |