Upload modelo treinado - 20251205_134240_hf

README.md

@@ -19,15 +19,15 @@ model-index:
       type: legal_documents
     metrics:
     - type: accuracy
-      value: 0.8856416772554002
+      value: 0.946376811594203
     - type: f1
-      value: 0.8776953377966165
+      value: 0.9406111634544171
     - type: precision
-      value: 0.8777274574075797
+      value: 0.9355706246585774
     - type: recall
-      value: 0.8856416772554002
+      value: 0.946376811594203
     - type: f1_macro
-      value: 0.3537898127235856
+      value: 0.7840758661419504
 ---
 
 # 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:** 23
+- **Número de Classes:** 6
 - **Idioma:** Português (Brasil)
 - **Domínio:** Documentos Jurídicos
 
@@ -63,65 +63,32 @@ Este modelo foi treinado para classificar documentos jurídicos brasileiros usan
 3. Citação
 4. Art.40
 5. Não
-6. Pelo prazo de 10 dias
-7. Sentença de mérito ou definitiva
-8. Citação de embargos
-9. Forma tácita
-10. Citação de penhora
-11. Art. 40
-12. sentença em apensado.
-13. Unificado
-14. Citação Negativa
-15. Arquivo provisório do art. 28 lef
-16. leilão
-17. intimação
-18. Citação de arresto
-19. Citação Positiva
-20. Penhora
-21. Desentramento
-22. Remessa para o arquivo
 
 ## Métricas de Avaliação
 
 As métricas abaixo foram calculadas no conjunto de validação:
 
-- **Accuracy:** 0.8856
-- **F1-Score (Weighted):** 0.8777
-- **Precision (Weighted):** 0.8777
-- **Recall (Weighted):** 0.8856
-- **F1-Score (Macro):** 0.3538
+- **Accuracy:** 0.9464
+- **F1-Score (Weighted):** 0.9406
+- **Precision (Weighted):** 0.9356
+- **Recall (Weighted):** 0.9464
+- **F1-Score (Macro):** 0.7841
 
 ## Classification Report Completo
 
 ```
               precision    recall  f1-score   support
 
-           0       0.50      0.14      0.22         7
-           1       0.94      0.96      0.95        51
-           2       1.00      0.86      0.92        49
-           3       0.80      0.88      0.84        42
-           4       0.94      0.96      0.95       138
-           5       0.96      0.96      0.96       414
-           6       0.00      0.00      0.00         7
-           7       1.00      1.00      1.00        16
-           8       0.00      0.00      0.00         4
-           9       0.19      1.00      0.32         6
-          10       0.00      0.00      0.00         2
-          11       0.91      1.00      0.95        10
-          12       0.33      0.62      0.43         8
-          13       0.21      0.27      0.24        11
-          14       0.00      0.00      0.00         3
-          15       0.00      0.00      0.00         2
-          16       0.00      0.00      0.00         1
-          17       0.00      0.00      0.00         6
-          19       0.00      0.00      0.00         3
-          20       0.00      0.00      0.00         5
-          21       0.00      0.00      0.00         1
-          22       0.00      0.00      0.00         1
-
-    accuracy                           0.89       787
-   macro avg       0.35      0.39      0.35       787
-weighted avg       0.88      0.89      0.88       787
+           0       0.00      0.00      0.00         8
+           1       0.92      1.00      0.96        45
+           2       0.94      0.85      0.89        54
+           3       0.93      0.95      0.94        41
+           4       0.94      0.98      0.96       121
+           5       0.96      0.96      0.96       421
+
+    accuracy                           0.95       690
+   macro avg       0.78      0.79      0.78       690
+weighted avg       0.94      0.95      0.94       690
 
 ```
 

config.json

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