Zip-1 / README.md
vdmbrsv's picture
Update README.md
3a9c699 verified
---
tags:
- sentence-transformers
- sentence-similarity
- feature-extraction
- tiny
base_model: sentence-transformers/all-MiniLM-L6-v2
pipeline_tag: sentence-similarity
library_name: sentence-transformers
metrics:
- pearson_cosine
- spearman_cosine
model-index:
- name: SentenceTransformer based on sentence-transformers/all-MiniLM-L6-v2
results:
- task:
type: semantic-similarity
name: Semantic Similarity
dataset:
name: Unknown
type: unknown
metrics:
- type: pearson_cosine
value: 0.6751697498221416
name: Pearson Cosine
- type: spearman_cosine
value: 0.7044137530273638
name: Spearman Cosine
---
# Super small embedding model (only 4MB!)
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("tabularisai/Zip-1")
# 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, 32]
# Get the similarity scores for the embeddings
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [3, 3]
print(similarities)
#tensor([[1.0000, 0.7272, 0.2864],
# [0.7272, 1.0000, 0.2265],
# [0.2864, 0.2265, 1.0000]])
```
<table align="center">
<tr>
<td align="center">
<a href="https://www.linkedin.com/company/tabularis-ai/">
<img src="https://cdn.jsdelivr.net/gh/simple-icons/simple-icons/icons/linkedin.svg" alt="LinkedIn" width="30" height="30">
</a>
</td>
<td align="center">
<a href="https://x.com/tabularis_ai">
<img src="https://cdn.jsdelivr.net/gh/simple-icons/simple-icons/icons/x.svg" alt="X" width="30" height="30">
</a>
</td>
<td align="center">
<a href="https://github.com/tabularis-ai">
<img src="https://cdn.jsdelivr.net/gh/simple-icons/simple-icons/icons/github.svg" alt="GitHub" width="30" height="30">
</a>
</td>
<td align="center">
<a href="https://tabularis.ai">
<img src="https://cdn.jsdelivr.net/gh/simple-icons/simple-icons/icons/internetarchive.svg" alt="Website" width="30" height="30">
</a>
</td>
</tr>
</table>