Sentence Similarity
sentence-transformers
PyTorch
TensorFlow
Rust
ONNX
Safetensors
OpenVINO
Transformers
albert
feature-extraction
Instructions to use sentence-transformers/paraphrase-albert-small-v2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use sentence-transformers/paraphrase-albert-small-v2 with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("sentence-transformers/paraphrase-albert-small-v2") sentences = [ "That is a happy person", "That is a happy dog", "That is a very happy person", "Today is a sunny day" ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4] - Transformers
How to use sentence-transformers/paraphrase-albert-small-v2 with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("sentence-transformers/paraphrase-albert-small-v2") model = AutoModel.from_pretrained("sentence-transformers/paraphrase-albert-small-v2", device_map="auto") - Inference
- Notebooks
- Google Colab
- Kaggle
Commit ·
b8a76dc
1
Parent(s): 9d5463a
adding training datasets
Browse files
README.md
CHANGED
|
@@ -7,9 +7,19 @@ tags:
|
|
| 7 |
- sentence-similarity
|
| 8 |
- transformers
|
| 9 |
datasets:
|
| 10 |
-
-
|
|
|
|
|
|
|
| 11 |
- wiki_atomic_edits
|
| 12 |
-
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 13 |
|
| 14 |
---
|
| 15 |
|
|
|
|
| 7 |
- sentence-similarity
|
| 8 |
- transformers
|
| 9 |
datasets:
|
| 10 |
+
- flax-sentence-embeddings/stackexchange_xml
|
| 11 |
+
- s2orc
|
| 12 |
+
- ms_marco
|
| 13 |
- wiki_atomic_edits
|
| 14 |
+
- snli
|
| 15 |
+
- multi_nli
|
| 16 |
+
- embedding-data/altlex
|
| 17 |
+
- embedding-data/simple-wiki
|
| 18 |
+
- embedding-data/flickr30k-captions
|
| 19 |
+
- embedding-data/coco_captions
|
| 20 |
+
- embedding-data/sentence-compression
|
| 21 |
+
- embedding-data/QQP
|
| 22 |
+
- yahoo_answers_topics
|
| 23 |
|
| 24 |
---
|
| 25 |
|