Views
No views yet
| Parameter | Value |
|---|---|
| Base model | neuralmind/bert-base-portuguese-cased |
| Loss | MultipleNegativesRankingLoss |
| Training pairs | 14,500 (adjacent same-section chunk pairs from 686 CVM filings) |
| Epochs | 10 |
| Batch size | 16 (effective 64 with gradient accumulation ×4) |
| Mixed precision | fp16 |
| Max sequence length | 256 tokens |
| Hardware | NVIDIA RTX A1000 (6 GB VRAM), ~2 hours |
| Initial loss | 2.201 (step 50) |
| Final loss | 0.115 (step 2,270) |
| Metric | Value |
|---|---|
| Recall@5 | 0.057 |
| Recall@10 | 0.071 |
| MRR | 0.100 |
| NDCG@10 | 0.063 |
1from sentence_transformers import SentenceTransformer
2
3model = SentenceTransformer("conderafael/cvm-bertimbau-sentence-transformer")
4
5# Encode a single passage
6embeddings = model.encode(["Receita líquida cresceu 12% no trimestre"])
7
8# Encode a batch
9texts = [
10 "O EBITDA ajustado atingiu R$ 4,2 bilhões no 3T24.",
11 "A Companhia mantém posição conservadora de hedge cambial.",
12]
13embeddings = model.encode(texts, normalize_embeddings=True)
14print(embeddings.shape) # (2, 768)