MultiPRIDE-DualEncoder-MainStage-it

README.md

@@ -21,11 +21,11 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [nickprock/setfit-italian-hate-speech](https://huggingface.co/nickprock/setfit-italian-hate-speech) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.3345
-- Accuracy: 0.9141
-- F1: 0.7407
-- Precision: 0.8696
-- Recall: 0.6452
+- Loss: 0.1685
+- Accuracy: 0.9571
+- F1: 0.8814
+- Precision: 0.9286
+- Recall: 0.8387
 
 ## Model description
 
@@ -47,7 +47,7 @@ The following hyperparameters were used during training:
 - learning_rate: 2e-05
 - train_batch_size: 8
 - eval_batch_size: 8
-- seed: 150
+- seed: 1337
 - optimizer: Use adamw_torch_fused with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
 - lr_scheduler_type: linear
 - num_epochs: 10
@@ -56,13 +56,16 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1     | Precision | Recall |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
-| 0.7188        | 1.0   | 95   | 0.5881          | 0.8405   | 0.4091 | 0.6923    | 0.2903 |
-| 0.5805        | 2.0   | 190  | 0.4346          | 0.8896   | 0.6667 | 0.7826    | 0.5806 |
-| 0.4433        | 3.0   | 285  | 0.3493          | 0.9141   | 0.7586 | 0.8148    | 0.7097 |
-| 0.3152        | 4.0   | 380  | 0.3066          | 0.9202   | 0.7797 | 0.8214    | 0.7419 |
-| 0.2758        | 5.0   | 475  | 0.3260          | 0.9080   | 0.7170 | 0.8636    | 0.6129 |
-| 0.2529        | 6.0   | 570  | 0.3228          | 0.9080   | 0.7273 | 0.8333    | 0.6452 |
-| 0.157         | 7.0   | 665  | 0.3345          | 0.9141   | 0.7407 | 0.8696    | 0.6452 |
+| 0.6452        | 1.0   | 95   | 0.5242          | 0.8344   | 0.6197 | 0.55      | 0.7097 |
+| 0.5208        | 2.0   | 190  | 0.3521          | 0.8896   | 0.7273 | 0.6857    | 0.7742 |
+| 0.4215        | 3.0   | 285  | 0.2637          | 0.9325   | 0.8136 | 0.8571    | 0.7742 |
+| 0.3325        | 4.0   | 380  | 0.2138          | 0.9325   | 0.8136 | 0.8571    | 0.7742 |
+| 0.2774        | 5.0   | 475  | 0.1805          | 0.9448   | 0.8525 | 0.8667    | 0.8387 |
+| 0.2297        | 6.0   | 570  | 0.1846          | 0.9509   | 0.8621 | 0.9259    | 0.8065 |
+| 0.3021        | 7.0   | 665  | 0.1664          | 0.9509   | 0.8667 | 0.8966    | 0.8387 |
+| 0.2372        | 8.0   | 760  | 0.1697          | 0.9571   | 0.8814 | 0.9286    | 0.8387 |
+| 0.2468        | 9.0   | 855  | 0.1688          | 0.9571   | 0.8814 | 0.9286    | 0.8387 |
+| 0.2456        | 10.0  | 950  | 0.1685          | 0.9571   | 0.8814 | 0.9286    | 0.8387 |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2fcd443b875f7ed98fb3f511f70d58feb54c707a914fe19dfd5e0299b3a22646
+oid sha256:770d78559a488fdbc295e35d023e4d6654287d67a7bab5396b54018e73891ac2
 size 443306324

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:acb67de67b6b83246de4618464653c099bd4268fdf962435c23a8e9fb63f387d
+oid sha256:f5a01c27a26bdfb27de2f89fff3fc9e35bc8cdfb681cd50d637aab5a4e2de4ea
 size 5969