Training 20251210_141008 - 7 classes with metrics

README.md

@@ -19,15 +19,15 @@ model-index:
       type: legal_documents
     metrics:
     - type: accuracy
-      value: 0.936231884057971
+      value: 0.9140401146131805
     - type: f1
-      value: 0.9353615414017292
+      value: 0.8994284142023327
     - type: precision
-      value: 0.9369069235885541
+      value: 0.8874266057437488
     - type: recall
-      value: 0.936231884057971
+      value: 0.9140401146131805
     - type: f1_macro
-      value: 0.8474441104841182
+      value: 0.6533238190381049
 ---
 
 # 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:** 6
+- **Número de Classes:** 7
 - **Idioma:** Português (Brasil)
 - **Domínio:** Documentos Jurídicos
 
@@ -63,32 +63,34 @@ Este modelo foi treinado para classificar documentos jurídicos brasileiros usan
 3. Citação
 4. Art. 40
 5. Não Influencia
+6. Citação de arresto
 
 ## Métricas de Avaliação
 
 As métricas abaixo foram calculadas no conjunto de validação:
 
-- **Accuracy:** 0.9362
-- **F1-Score (Weighted):** 0.9354
-- **Precision (Weighted):** 0.9369
-- **Recall (Weighted):** 0.9362
-- **F1-Score (Macro):** 0.8474
+- **Accuracy:** 0.9140
+- **F1-Score (Weighted):** 0.8994
+- **Precision (Weighted):** 0.8874
+- **Recall (Weighted):** 0.9140
+- **F1-Score (Macro):** 0.6533
 
 ## Classification Report Completo
 
 ```
               precision    recall  f1-score   support
 
-           0       0.60      0.38      0.46         8
-           1       0.86      0.98      0.92        45
-           2       1.00      0.85      0.92        54
-           3       0.85      0.95      0.90        41
-           4       0.93      0.95      0.94       121
-           5       0.95      0.95      0.95       421
-
-    accuracy                           0.94       690
-   macro avg       0.87      0.84      0.85       690
-weighted avg       0.94      0.94      0.94       690
+           0       0.00      0.00      0.00         8
+           1       0.98      0.96      0.97        46
+           2       0.93      0.76      0.84        55
+           3       0.85      0.97      0.91        36
+           4       0.90      0.94      0.92       120
+           5       0.92      0.96      0.94       420
+           6       0.00      0.00      0.00        13
+
+    accuracy                           0.91       698
+   macro avg       0.65      0.66      0.65       698
+weighted avg       0.89      0.91      0.90       698
 
 ```
 

config.json

@@ -15,7 +15,8 @@
     "2": "646bc57bc7605185f40b4263",
     "3": "646b78c3dc761bc8c743a0c6",
     "4": "645d42ebbacbc7baba32624d",
-    "5": "64ac68185fe31ca8ad57a822"
+    "5": "64ac68185fe31ca8ad57a822",
+    "6": "64ede93370af4a0007c2afb8"
   },
   "initializer_range": 0.02,
   "intermediate_size": 3072,
@@ -25,6 +26,7 @@
     "646bb41b351734d0a39bb557": 0,
     "646bc57bc7605185f40b4263": 2,
     "64ac68185fe31ca8ad57a822": 5,
+    "64ede93370af4a0007c2afb8": 6,
     "652026c3798f7600075f9d40": 1
   },
   "layer_norm_eps": 1e-12,
@@ -40,7 +42,7 @@
   "pooler_type": "first_token_transform",
   "position_embedding_type": "absolute",
   "problem_type": "single_label_classification",
-  "transformers_version": "4.57.1",
+  "transformers_version": "4.57.3",
   "type_vocab_size": 2,
   "use_cache": true,
   "vocab_size": 119547

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ebbc0bde22fe9605d6b416bb40abfd1a9c25bb23293ebd059249447c5a5db0fb
-size 711455760
+oid sha256:b6cb3412f6c9418d62ef36479d9d4387a880ed96f8c0dcca2de855d591fe9c5a
+size 711458836