PeppoCola commited on
Commit
7792eca
1 Parent(s): de6d637

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -2
README.md CHANGED
@@ -14,7 +14,8 @@ This is a [sentence-transformers](https://www.SBERT.net) model, specific for Git
14
 
15
  ## Dataset
16
 
17
- For training, we used the [NLBSE22 dataset](https://nlbse2022.github.io/tools/)
 
18
 
19
 
20
  ## Usage (Sentence-Transformers)
@@ -38,7 +39,8 @@ print(embeddings)
38
 
39
 
40
  ## Training
41
- The model was trained with the parameters:
 
42
 
43
  **DataLoader**:
44
 
 
14
 
15
  ## Dataset
16
 
17
+ For training, we used the [NLBSE22 dataset](https://nlbse2022.github.io/tools/), after removing issues with empty body and duplicates.
18
+ Similarity between title and body was used to train the sentence embedding model.
19
 
20
 
21
  ## Usage (Sentence-Transformers)
 
39
 
40
 
41
  ## Training
42
+ The model was trained for ten epochs, using Multiple Negative Ranking Loss. We assumed that title and body of the same issue have to be similar.
43
+ We used the following parameters:
44
 
45
  **DataLoader**:
46