Training 20251202_154101 - 7 classes with metrics

README.md

@@ -1,6 +1,33 @@
 ---
-library_name: transformers
-tags: []
+datasets:
+- legal_documents
+language: pt
+license: apache-2.0
+tags:
+- text-classification
+- legal
+- bert
+- portuguese
+- brazilian-legal-documents
+model-index:
+- name: testmodel
+  results:
+  - task:
+      type: text-classification
+    dataset:
+      name: Legal Documents Classification Dataset
+      type: legal_documents
+    metrics:
+    - type: accuracy
+      value: 0.941970310391363
+    - type: f1
+      value: 0.9370428524548865
+    - type: precision
+      value: 0.9343445431988082
+    - type: recall
+      value: 0.941970310391363
+    - type: f1_macro
+      value: 0.7992595460988434
 ---
 
 # Model Card for Model ID
@@ -15,14 +42,100 @@ tags: []
 
 <!-- Provide a longer summary of what this model is. -->
 
-This is the model card of a 🤗 transformers model that has been pushed on the Hub. This model card has been automatically generated.
+
+# Modelo de Classificação de Documentos Jurídicos
+
+Este modelo foi treinado para classificar documentos jurídicos brasileiros usando BERT multilíngue.
+
+## Detalhes do Modelo
+
+- **Base Model:** google-bert/bert-base-multilingual-cased
+- **Tipo:** Classificação Multiclasse
+- **Número de Classes:** 7
+- **Idioma:** Português (Brasil)
+- **Domínio:** Documentos Jurídicos
+
+## Classes
+
+0. Alvará
+1. Sentença
+2. Penhora/Arresto/Sequestro
+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
+
+## 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
+
+## Classification Report Completo
+
+```
+              precision    recall  f1-score   support
+
+           0       0.00      0.00      0.00         6
+           1       0.94      0.96      0.95        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
+           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
+
+```
+
+## Uso
+
+```python
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+import torch
+
+# Carregar modelo e tokenizer
+model = AutoModelForSequenceClassification.from_pretrained("rkenji/testmodel")
+tokenizer = AutoTokenizer.from_pretrained("rkenji/testmodel")
+
+# Fazer predição
+texto = "Seu texto jurídico aqui"
+inputs = tokenizer(texto, return_tensors="pt", truncation=True, max_length=512)
+outputs = model(**inputs)
+prediction = torch.argmax(outputs.logits, dim=1).item()
+
+print(f"Classe predita: {prediction}")
+```
+
+## Treinamento
+
+O modelo foi treinado com early stopping e validação cruzada em um dataset de documentos jurídicos brasileiros.
+
 
 - **Developed by:** [More Information Needed]
 - **Funded by [optional]:** [More Information Needed]
 - **Shared by [optional]:** [More Information Needed]
 - **Model type:** [More Information Needed]
-- **Language(s) (NLP):** [More Information Needed]
-- **License:** [More Information Needed]
+- **Language(s) (NLP):** pt
+- **License:** apache-2.0
 - **Finetuned from model [optional]:** [More Information Needed]
 
 ### Model Sources [optional]

config.json

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