Update README.md
Browse files
README.md
CHANGED
@@ -25,6 +25,19 @@ This is a [SetFit model](https://github.com/huggingface/setfit) that can be used
|
|
25 |
1. Fine-tuning a [Sentence Transformer](https://www.sbert.net) with contrastive learning.
|
26 |
2. Training a classification head with features from the fine-tuned Sentence Transformer.
|
27 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
28 |
## Usage
|
29 |
|
30 |
To use this model for inference, first install the SetFit library:
|
|
|
25 |
1. Fine-tuning a [Sentence Transformer](https://www.sbert.net) with contrastive learning.
|
26 |
2. Training a classification head with features from the fine-tuned Sentence Transformer.
|
27 |
|
28 |
+
## Train Hyperparameters
|
29 |
+
|
30 |
+
* Simulate the few-shot regime by sampling 25 examples per class
|
31 |
+
* Sentence Transformer checkpoint: *"sentence-transformers/paraphrase-distilroberta-base-v2"*
|
32 |
+
* Number of text pairs to generate for contrastive learning: 10
|
33 |
+
* Epochs: 1
|
34 |
+
* Batch size: 32
|
35 |
+
|
36 |
+
## Metrics on Evaluation set
|
37 |
+
|
38 |
+
* accuracy score: 0.8529
|
39 |
+
* f1 score: 0.8527
|
40 |
+
|
41 |
## Usage
|
42 |
|
43 |
To use this model for inference, first install the SetFit library:
|