Update README.md
Browse files
README.md
CHANGED
|
@@ -44,7 +44,14 @@ datasets:
|
|
| 44 |
- **Batch Size**: 64
|
| 45 |
- **Epochs**: 200
|
| 46 |
- **Loss Function**: Focal Loss to handle class imbalance
|
| 47 |
-
- **Frameworks
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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:
|
|
|
|
| 44 |
- **Batch Size**: 64
|
| 45 |
- **Epochs**: 200
|
| 46 |
- **Loss Function**: Focal Loss to handle class imbalance
|
| 47 |
+
- **Frameworks**
|
| 48 |
+
: PyTorch, Hugging Face Transformers, SimpleTransformers
|
| 49 |
+
|
| 50 |
+
## Evaluation metrics
|
| 51 |
+
- eval_loss = 0.3708431158236593
|
| 52 |
+
- f1_score = 0.7571850298211653
|
| 53 |
+
- precision = 0.7577626541897065
|
| 54 |
+
- recall = 0.7566082854003748
|
| 55 |
|
| 56 |
## How to Use
|
| 57 |
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:
|