Training 20251203_131528 - 13 classes with metrics

README.md

@@ -19,15 +19,15 @@ model-index:
       type: legal_documents
     metrics:
     - type: accuracy
-      value: 0.9352226720647774
+      value: 0.9230769230769231
     - type: f1
-      value: 0.9328717264101649
+      value: 0.9213415721061173
     - type: precision
-      value: 0.9342469820430453
+      value: 0.9231096484224747
     - type: recall
-      value: 0.9352226720647774
+      value: 0.9230769230769231
     - type: f1_macro
-      value: 0.8259694427208276
+      value: 0.4819512053123249
 ---
 
 # Model Card for Model ID
@@ -51,7 +51,7 @@ Este modelo foi treinado para classificar documentos jurídicos brasileiros usan
 
 - **Base Model:** google-bert/bert-base-multilingual-cased
 - **Tipo:** Classificação Multiclasse
-- **Número de Classes:** 7
+- **Número de Classes:** 13
 - **Idioma:** Português (Brasil)
 - **Domínio:** Documentos Jurídicos
 
@@ -63,46 +63,51 @@ Este modelo foi treinado para classificar documentos jurídicos brasileiros usan
 3. Citação
 4. Art.40
 5. Não
-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
+6. Alvará
+7. Citação
+8. sentença em apensado.
+9. Sentença de mérito ou definitiva
+10. Art. 40
+11. Citação de embargos
+12. Volta da suspensão do artigo 40
+13. Unificado
+14. Arquivo provisório do art. 28 lef
+15. Remessa para o arquivo
+16. Pelo prazo de 10 dias
+17. leilão
+18. Forma tácita
 
 ## Métricas de Avaliação
 
 As métricas abaixo foram calculadas no conjunto de validação:
 
-- **Accuracy:** 0.9352
-- **F1-Score (Weighted):** 0.9329
-- **Precision (Weighted):** 0.9342
-- **Recall (Weighted):** 0.9352
-- **F1-Score (Macro):** 0.8260
+- **Accuracy:** 0.9231
+- **F1-Score (Weighted):** 0.9213
+- **Precision (Weighted):** 0.9231
+- **Recall (Weighted):** 0.9231
+- **F1-Score (Macro):** 0.4820
 
 ## Classification Report Completo
 
 ```
               precision    recall  f1-score   support
 
-           0       0.33      0.17      0.22         6
-           1       0.86      0.96      0.91        46
-           2       0.98      0.73      0.83        55
-           3       0.89      0.97      0.93        35
-           4       0.92      0.95      0.94       132
-           5       0.95      0.96      0.95       424
-           6       1.00      1.00      1.00        43
-
-    accuracy                           0.94       741
-   macro avg       0.85      0.82      0.83       741
-weighted avg       0.93      0.94      0.93       741
+           0       0.60      0.38      0.46         8
+           1       0.90      0.94      0.92        47
+           2       0.96      0.82      0.88        61
+           3       0.90      0.98      0.94        45
+           4       0.91      0.97      0.94       140
+           5       0.96      0.95      0.95       427
+           7       0.00      0.00      0.00         1
+           8       0.29      1.00      0.44         2
+           9       0.00      0.00      0.00         1
+          10       0.33      0.20      0.25         5
+          11       0.00      0.00      0.00         2
+          12       0.00      0.00      0.00         2
+
+    accuracy                           0.92       741
+   macro avg       0.49      0.52      0.48       741
+weighted avg       0.92      0.92      0.92       741
 
 ```
 

config.json

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