Update README.md
Browse files
README.md
CHANGED
|
@@ -38,8 +38,8 @@ Use the code below to get started with the model.
|
|
| 38 |
# Load model directly
|
| 39 |
from transformers import AutoTokenizer, TFAutoModelForQuestionAnswering
|
| 40 |
|
| 41 |
-
tokenizer = AutoTokenizer.from_pretrained("Tien-THM/
|
| 42 |
-
model = TFAutoModelForQuestionAnswering.from_pretrained("Tien-THM/
|
| 43 |
|
| 44 |
import numpy as np
|
| 45 |
|
|
@@ -79,9 +79,6 @@ Using 2 datasets:
|
|
| 79 |
* Learning rate: 2e-5
|
| 80 |
* Batch size: 8
|
| 81 |
* Epoch: 4
|
| 82 |
-
'train_loss': [4.7110, 3.2650, 2.7899, 2.4633],
|
| 83 |
-
'val_loss': [3.6251, 3.3062, 3.2184, 3.1946],
|
| 84 |
-
'exact_match': [0.38, 0.42, 0.44, 0.45]
|
| 85 |
#### Training Loss
|
| 86 |
| Epoch | Train loss | Validation loss | Exact Match |
|
| 87 |
|----------|----------|----------|----------|
|
|
|
|
| 38 |
# Load model directly
|
| 39 |
from transformers import AutoTokenizer, TFAutoModelForQuestionAnswering
|
| 40 |
|
| 41 |
+
tokenizer = AutoTokenizer.from_pretrained("Tien-THM/bert-mini-fine-tuning-squad")
|
| 42 |
+
model = TFAutoModelForQuestionAnswering.from_pretrained("Tien-THM/bert-mini-fine-tuning-squad")
|
| 43 |
|
| 44 |
import numpy as np
|
| 45 |
|
|
|
|
| 79 |
* Learning rate: 2e-5
|
| 80 |
* Batch size: 8
|
| 81 |
* Epoch: 4
|
|
|
|
|
|
|
|
|
|
| 82 |
#### Training Loss
|
| 83 |
| Epoch | Train loss | Validation loss | Exact Match |
|
| 84 |
|----------|----------|----------|----------|
|