Update README.md

README.md

@@ -150,9 +150,41 @@ result = citation_classifier(citation_text)
 print(result)
 ```
 
+## Training
+
+The model was trained using the **Citation Pre-Screening Dataset** consisting of:
+
+- **Training data**: 3599 samples
+- **Test data**: 400 samples
+
+The following hyperparameters were used for training:
+
+- **Model Path**: `distilbert/distilbert-base-multilingual-cased`
+- **Batch Size**: 32
+- **Number of Epochs**: 4
+- **Learning Rate**: 2e-5
+- **Max Sequence Length**: 512
+
+## Evaluation Metrics
+
+The model's performance was evaluated on the test set, and the following results were obtained:
+
+| Metric               | Value  |
+|----------------------|--------|
+| **Accuracy**          | 0.95   |
+| **Macro avg F1**      | 0.94   |
+| **Weighted avg F1**   | 0.95   |
 
 ## Additional information
 
 ### Authors 
 
 - SIRIS Lab, Research Division of SIRIS Academic.
+
+### License
+
+This work is distributed under a [Apache License, Version 2.0](https://www.apache.org/licenses/LICENSE-2.0).
+
+### Contact
+For further information, send an email to either [nicolau.duransilva@sirisacademic.com](mailto:nicolau.duransilva@sirisacademic.com) or [info@sirisacademic.com](mailto:info@sirisacademic.com).
+