Training 20251203_124401 - 7 classes with metrics

README.md

@@ -19,15 +19,15 @@ model-index:
       type: legal_documents
     metrics:
     - type: accuracy
-      value: 0.941970310391363
+      value: 0.9352226720647774
     - type: f1
-      value: 0.9370428524548865
+      value: 0.9301815453882951
     - type: precision
-      value: 0.9343445431988082
+      value: 0.9296807015714141
     - type: recall
-      value: 0.941970310391363
+      value: 0.9352226720647774
     - type: f1_macro
-      value: 0.7992595460988434
+      value: 0.7944170155534728
 ---
 
 # Model Card for Model ID
@@ -63,29 +63,29 @@ Este modelo foi treinado para classificar documentos jurídicos brasileiros usan
 3. Citação
 4. Art.40
 5. Não
-6. Unificado
-7. sentença em apensado.
-8. Arquivo provisório do art. 28 lef
-9. Alvará
-10. Citação de embargos
-11. Remessa para o arquivo
-12. Sentença de mérito ou definitiva
-13. Citação
-14. Volta da suspensão do artigo 40
-15. Forma tácita
-16. Pelo prazo de 10 dias
-17. leilão
-18. Art. 40
+6. Sentença de mérito ou definitiva
+7. Remessa para o arquivo
+8. Unificado
+9. Volta da suspensão do artigo 40
+10. Citação
+11. sentença em apensado.
+12. Forma tácita
+13. leilão
+14. Pelo prazo de 10 dias
+15. Arquivo provisório do art. 28 lef
+16. Alvará
+17. Art. 40
+18. Citação de embargos
 
 ## Métricas de Avaliação
 
 As métricas abaixo foram calculadas no conjunto de validação:
 
-- **Accuracy:** 0.9420
-- **F1-Score (Weighted):** 0.9370
-- **Precision (Weighted):** 0.9343
-- **Recall (Weighted):** 0.9420
-- **F1-Score (Macro):** 0.7993
+- **Accuracy:** 0.9352
+- **F1-Score (Weighted):** 0.9302
+- **Precision (Weighted):** 0.9297
+- **Recall (Weighted):** 0.9352
+- **F1-Score (Macro):** 0.7944
 
 ## Classification Report Completo
 
@@ -93,16 +93,16 @@ As métricas abaixo foram calculadas no conjunto de validação:
               precision    recall  f1-score   support
 
            0       0.00      0.00      0.00         6
-           1       0.94      0.96      0.95        46
+           1       0.88      0.93      0.91        46
            2       0.93      0.73      0.82        55
-           3       0.89      0.97      0.93        35
-           4       0.93      0.95      0.94       132
-           5       0.94      0.97      0.96       424
+           3       0.97      0.94      0.96        35
+           4       0.99      0.88      0.93       132
+           5       0.92      0.99      0.95       424
            6       1.00      1.00      1.00        43
 
     accuracy                           0.94       741
-   macro avg       0.81      0.80      0.80       741
-weighted avg       0.93      0.94      0.94       741
+   macro avg       0.81      0.78      0.79       741
+weighted avg       0.93      0.94      0.93       741
 
 ```
 

config.json

@@ -16,40 +16,40 @@
     "3": "Cita\u00e7\u00e3o",
     "4": "Art.40",
     "5": "N\u00e3o",
-    "6": "Unificado",
-    "7": "senten\u00e7a em apensado.",
-    "8": "Arquivo provis\u00f3rio do art. 28 lef",
-    "9": "Alvar\u00e1",
-    "10": "Cita\u00e7\u00e3o de embargos",
-    "11": "Remessa para o arquivo",
-    "12": "Senten\u00e7a de m\u00e9rito ou definitiva",
-    "13": "Cita\u00e7\u00e3o",
-    "14": "Volta da suspens\u00e3o do artigo 40",
-    "15": "Forma t\u00e1cita",
-    "16": "Pelo prazo de 10 dias",
-    "17": "leil\u00e3o",
-    "18": "Art. 40"
+    "6": "Senten\u00e7a de m\u00e9rito ou definitiva",
+    "7": "Remessa para o arquivo",
+    "8": "Unificado",
+    "9": "Volta da suspens\u00e3o do artigo 40",
+    "10": "Cita\u00e7\u00e3o",
+    "11": "senten\u00e7a em apensado.",
+    "12": "Forma t\u00e1cita",
+    "13": "leil\u00e3o",
+    "14": "Pelo prazo de 10 dias",
+    "15": "Arquivo provis\u00f3rio do art. 28 lef",
+    "16": "Alvar\u00e1",
+    "17": "Art. 40",
+    "18": "Cita\u00e7\u00e3o de embargos"
   },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "Alvar\u00e1": 9,
-    "Arquivo provis\u00f3rio do art. 28 lef": 8,
-    "Art. 40": 18,
+    "Alvar\u00e1": 16,
+    "Arquivo provis\u00f3rio do art. 28 lef": 15,
+    "Art. 40": 17,
     "Art.40": 4,
-    "Cita\u00e7\u00e3o": 13,
-    "Cita\u00e7\u00e3o de embargos": 10,
-    "Forma t\u00e1cita": 15,
+    "Cita\u00e7\u00e3o": 10,
+    "Cita\u00e7\u00e3o de embargos": 18,
+    "Forma t\u00e1cita": 12,
     "N\u00e3o": 5,
-    "Pelo prazo de 10 dias": 16,
+    "Pelo prazo de 10 dias": 14,
     "Penhora/Arresto/Sequestro": 2,
-    "Remessa para o arquivo": 11,
+    "Remessa para o arquivo": 7,
     "Senten\u00e7a": 1,
-    "Senten\u00e7a de m\u00e9rito ou definitiva": 12,
-    "Unificado": 6,
-    "Volta da suspens\u00e3o do artigo 40": 14,
-    "leil\u00e3o": 17,
-    "senten\u00e7a em apensado.": 7
+    "Senten\u00e7a de m\u00e9rito ou definitiva": 6,
+    "Unificado": 8,
+    "Volta da suspens\u00e3o do artigo 40": 9,
+    "leil\u00e3o": 13,
+    "senten\u00e7a em apensado.": 11
   },
   "layer_norm_eps": 1e-12,
   "max_position_embeddings": 512,