nickprock commited on
Commit
4737732
1 Parent(s): f6ddba2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +13 -0
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: