Update README.md
Browse files
README.md
CHANGED
|
@@ -38,6 +38,28 @@ widget:
|
|
| 38 |
'Art':7,
|
| 39 |
'Accidents':8
|
| 40 |
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 41 |
|
| 42 |
# Example usage
|
| 43 |
```python
|
|
|
|
| 38 |
'Art':7,
|
| 39 |
'Accidents':8
|
| 40 |
}
|
| 41 |
+
## Training parameters
|
| 42 |
+
|
| 43 |
+
| | |
|
| 44 |
+
| :-------------------: | :-----------:|
|
| 45 |
+
| Training batch size | `8` |
|
| 46 |
+
| Evaluation batch size | `8` |
|
| 47 |
+
| learning rate | `1e-4` |
|
| 48 |
+
| Max length input | `64` |
|
| 49 |
+
| Max length target | `512` |
|
| 50 |
+
| Number workers | `4` |
|
| 51 |
+
|
| 52 |
+
|
| 53 |
+
## Results
|
| 54 |
+
|
| 55 |
+
| | |
|
| 56 |
+
| :---------------------: | :-----------: |
|
| 57 |
+
| Evaluation Loss | `3.94` |
|
| 58 |
+
| Training Loss | `3.20` |
|
| 59 |
+
| Classification accuracy | `56.84%` |
|
| 60 |
+
| Generation accuracy | `56.84%` |
|
| 61 |
+
| ClassificationBLEU | `56.84%` |
|
| 62 |
+
| Generation BLEU | `56.84%` |
|
| 63 |
|
| 64 |
# Example usage
|
| 65 |
```python
|