Update README.md
Browse files
README.md
CHANGED
|
@@ -38,7 +38,8 @@ tags:
|
|
| 38 |
- **Test Set**: 10%
|
| 39 |
|
| 40 |
## Model Training
|
| 41 |
-
-
|
|
|
|
| 42 |
- **Optimizer**: AdamW
|
| 43 |
- **Learning Rate**: 3e-5
|
| 44 |
- **Batch Size**: 64
|
|
@@ -46,6 +47,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 |
|
|
|
|
| 38 |
- **Test Set**: 10%
|
| 39 |
|
| 40 |
## Model Training
|
| 41 |
+
-
|
| 42 |
+
**Training Configuration**:
|
| 43 |
- **Optimizer**: AdamW
|
| 44 |
- **Learning Rate**: 3e-5
|
| 45 |
- **Batch Size**: 64
|
|
|
|
| 47 |
- **Loss Function**: Focal Loss to handle class imbalance
|
| 48 |
- **Frameworks**: PyTorch, Hugging Face Transformers, SimpleTransformers
|
| 49 |
|
| 50 |
+
## Evaluation metrics
|
| 51 |
+
- eval_loss = 0.4112480320792267
|
| 52 |
+
- f1_score = 0.6910085729376871
|
| 53 |
+
- precision = 0.7068717096148518
|
| 54 |
+
- recall = 0.675841788751424
|
| 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:
|
| 58 |
|