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 |
|