Update README.md
Browse files
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
|
|
|
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 |
|