Feature Extraction
Transformers
Safetensors
English
bert
tadse
sentence-embeddings
contrastive-learning
model-merge
text-embeddings-inference
Instructions to use minsik-oh/template-dialogue-embedding with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use minsik-oh/template-dialogue-embedding with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("feature-extraction", model="minsik-oh/template-dialogue-embedding")# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("minsik-oh/template-dialogue-embedding") model = AutoModel.from_pretrained("minsik-oh/template-dialogue-embedding", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Download evaluation.json from minsik-oh/template-dialogue-embedding: direct link, hf CLI and curl.
- Browser
- Download file 1.17 kB
-
https://huggingface.co/minsik-oh/template-dialogue-embedding/resolve/main/evaluation.json
- Command line
-
hf download hf://minsik-oh/template-dialogue-embedding/evaluation.json
-
curl -L -o evaluation.json https://huggingface.co/minsik-oh/template-dialogue-embedding/resolve/main/evaluation.json
1.17 kB
| { | |
| "protocol": "recovered-48d5f94-utterance-only", | |
| "settings": { | |
| "pooler": "cls", | |
| "max_length": 128, | |
| "batch_size": 128, | |
| "aux_ratio": 0, | |
| "search_target": "train", | |
| "utterance_topk": 1 | |
| }, | |
| "mean_accuracy": 83.96792567576395, | |
| "simcse_mean": 81.30318600054206, | |
| "tasks": { | |
| "snips": { | |
| "accuracy": 96.0, | |
| "correct": 672, | |
| "test_count": 700, | |
| "reference_count": 13784, | |
| "simcse_accuracy": 92.0 | |
| }, | |
| "atis": { | |
| "accuracy": 89.58566629339306, | |
| "correct": 800, | |
| "test_count": 893, | |
| "reference_count": 4978, | |
| "simcse_accuracy": 85.66629339305712 | |
| }, | |
| "massive": { | |
| "accuracy": 79.58977807666443, | |
| "correct": 2367, | |
| "test_count": 2974, | |
| "reference_count": 11514, | |
| "simcse_accuracy": 76.76529926025555 | |
| }, | |
| "hwu": { | |
| "accuracy": 82.40963855421687, | |
| "correct": 684, | |
| "test_count": 830, | |
| "reference_count": 19229, | |
| "simcse_accuracy": 81.0843373493976 | |
| }, | |
| "clinc": { | |
| "accuracy": 72.25454545454545, | |
| "correct": 3974, | |
| "test_count": 5500, | |
| "reference_count": 15250, | |
| "simcse_accuracy": 71.0 | |
| } | |
| } | |
| } | |