Update README.md
Browse files
README.md
CHANGED
|
@@ -43,7 +43,8 @@ PROBLEM`: Diseases, symptoms, and medical conditions.
|
|
| 43 |
- **Optimizer**: AdamW
|
| 44 |
- **Learning Rate**: 3e-5
|
| 45 |
- **Batch Size**: 64
|
| 46 |
-
-
|
|
|
|
| 47 |
- **Loss Function**: Focal Loss to handle class imbalance
|
| 48 |
- **Frameworks**: PyTorch, Hugging Face Transformers, SimpleTransformers
|
| 49 |
|
|
@@ -58,7 +59,6 @@ You can easily use this model with the Hugging Face `transformers` library. Here
|
|
| 58 |
|
| 59 |
```python
|
| 60 |
from transformers import AutoTokenizer, AutoModelForTokenClassification
|
| 61 |
-
import torch
|
| 62 |
|
| 63 |
model_name = "rigonsallauka/polish_medical_ner"
|
| 64 |
|
|
|
|
| 43 |
- **Optimizer**: AdamW
|
| 44 |
- **Learning Rate**: 3e-5
|
| 45 |
- **Batch Size**: 64
|
| 46 |
+
-
|
| 47 |
+
**Epochs**: 200
|
| 48 |
- **Loss Function**: Focal Loss to handle class imbalance
|
| 49 |
- **Frameworks**: PyTorch, Hugging Face Transformers, SimpleTransformers
|
| 50 |
|
|
|
|
| 59 |
|
| 60 |
```python
|
| 61 |
from transformers import AutoTokenizer, AutoModelForTokenClassification
|
|
|
|
| 62 |
|
| 63 |
model_name = "rigonsallauka/polish_medical_ner"
|
| 64 |
|