Views
No views yet
neuralmind/bert-base-portuguese-cased1Sentence A: "Pelo Senhor Presidente foi presente a reunião a ata n.º 28 de 20.12.2023."
2Sentence B: "Ponderado e analisado o assunto o Executivo Municipal deliberou por unanimidade aprovar a ata n.º 28 de 20.12.2023."
3→ Prediction: Same Topic (confidence: 76%)
4
5Sentence A: "Ponderado e analisado o assunto o Executivo Municipal deliberou por unanimidade aprovar a ata n.º 28 de 20.12.2023."
6Sentence B: "Não houve processos e requerimentos diversos a apresentar."
7→ Prediction: Topic Boundary (confidence: 82%)1from transformers import AutoTokenizer, AutoModelForNextSentencePrediction
2import torch
3
4# Load model and tokenizer
5tokenizer = AutoTokenizer.from_pretrained("liaad/Citilink-NSP-Segmentation-pt")
6model = AutoModelForNextSentencePrediction.from_pretrained("liaad/Citilink-NSP-Segmentation-pt")
7
8# Prepare input
9sentence_a = "Pelo Senhor Presidente foi presente a reunião a ata n.º 28 de 20.12.2023."
10sentence_b = "Ponderado e analisado o assunto o Executivo Municipal deliberou por unanimidade aprovar a ata n.º 28 de 20.12.2023."
11
12
13# Tokenize
14inputs = tokenizer(sentence_a, sentence_b, return_tensors="pt")
15
16# Predict
17with torch.no_grad():
18 outputs = model(**inputs)
19 logits = outputs.logits
20 probs = torch.softmax(logits, dim=1)
21
22# Interpret results
23is_next_prob = probs[0][0].item()
24not_next_prob = probs[0][1].item()
25
26print(f"Is Next (same topic): {is_next_prob:.3f}")
27print(f"Not Next (topic boundary): {not_next_prob:.3f}")
28
29if not_next_prob > 0.5:
30 print("🔴 Topic boundary detected!")
31else:
32 print("🟢 Same topic continues")