Update README.md
Browse files
README.md
CHANGED
|
@@ -46,6 +46,12 @@ tags:
|
|
| 46 |
- **Loss Function**: Focal Loss to handle class imbalance
|
| 47 |
- **Frameworks**: PyTorch, Hugging Face Transformers, SimpleTransformers
|
| 48 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 49 |
## How to Use
|
| 50 |
You can easily use this model with the Hugging Face `transformers` library. Here's an example of how to load and use the model for inference:
|
| 51 |
|
|
|
|
| 46 |
- **Loss Function**: Focal Loss to handle class imbalance
|
| 47 |
- **Frameworks**: PyTorch, Hugging Face Transformers, SimpleTransformers
|
| 48 |
|
| 49 |
+
## Evaluation metrics
|
| 50 |
+
- eval_loss = 0.3371218325682951
|
| 51 |
+
- f1_score = 0.7559515712148007
|
| 52 |
+
- precision = 0.759089632772006
|
| 53 |
+
- recall = 0.7528393482105897
|
| 54 |
+
|
| 55 |
## How to Use
|
| 56 |
You can easily use this model with the Hugging Face `transformers` library. Here's an example of how to load and use the model for inference:
|
| 57 |
|