Update README.md
Browse files
README.md
CHANGED
|
@@ -156,9 +156,11 @@ You can finetune this model on your own dataset.
|
|
| 156 |
|
| 157 |
### Training Dataset
|
| 158 |
|
| 159 |
-
####
|
|
|
|
|
|
|
| 160 |
|
| 161 |
-
* Dataset:
|
| 162 |
* Size: 300,000 training samples
|
| 163 |
* Columns: <code>sentence_a</code>, <code>sentence_b</code>, and <code>label</code>
|
| 164 |
* Approximate statistics based on the first 1000 samples:
|
|
@@ -307,7 +309,7 @@ You can finetune this model on your own dataset.
|
|
| 307 |
### Training Logs
|
| 308 |
<details><summary>Click to expand</summary>
|
| 309 |
|
| 310 |
-
| Epoch | Step | Training Loss |
|
| 311 |
|:------:|:-----:|:-------------:|:--------------------:|:----------------------:|:-----------------------:|
|
| 312 |
| 0.0026 | 10 | 1.1707 | - | - | - |
|
| 313 |
| 0.0051 | 20 | 1.4398 | - | - | - |
|
|
|
|
| 156 |
|
| 157 |
### Training Dataset
|
| 158 |
|
| 159 |
+
#### https://huggingface.co/datasets/MasterControlAIML/triplet_finetuning_embeddings.csv
|
| 160 |
+
#### https://huggingface.co/datasets/MasterControlAIML/pair_class_finetuning_embeddings.csv
|
| 161 |
+
#### https://huggingface.co/datasets/MasterControlAIML/question_answer_finetuning_embeddings.csv
|
| 162 |
|
| 163 |
+
* Dataset: MasterControlAIML/pair_class_finetuning_embeddings.csv
|
| 164 |
* Size: 300,000 training samples
|
| 165 |
* Columns: <code>sentence_a</code>, <code>sentence_b</code>, and <code>label</code>
|
| 166 |
* Approximate statistics based on the first 1000 samples:
|
|
|
|
| 309 |
### Training Logs
|
| 310 |
<details><summary>Click to expand</summary>
|
| 311 |
|
| 312 |
+
| Epoch | Step | Training Loss | triplet loss | natural-questions loss | pair-class loss |
|
| 313 |
|:------:|:-----:|:-------------:|:--------------------:|:----------------------:|:-----------------------:|
|
| 314 |
| 0.0026 | 10 | 1.1707 | - | - | - |
|
| 315 |
| 0.0051 | 20 | 1.4398 | - | - | - |
|