Views
No views yet
bert-base-uncased optimized specifically for identifying binary classification boundaries between legitimate business communication and malicious phishing/fraudulent email templates.NaN log updates) found in larger architectures by utilizing strict token length packing bounds and zero-division safeguards during training evaluation.| Metric | Final Score |
|---|---|
| Accuracy | 99.22% |
| F1-Score | 99.25% |
| Precision | 99.77% |
| Recall | 98.73% |
| Final Training Loss | 0.0408 |
transformers framework:1import torch
2import numpy as np
3from transformers import AutoTokenizer, AutoModelForSequenceClassification
4
5model_id = "sourav-paramanya/bert-base-bank-mail-fraud-detector"
6tokenizer = AutoTokenizer.from_pretrained(model_id)
7model = AutoModelForSequenceClassification.from_pretrained(model_id)
8
9def scan_incoming_email(email_text):
10 inputs = tokenizer(email_text, return_tensors="pt", padding=True, truncation=True, max_length=512)
11
12 with torch.no_grad():
13 outputs = model(**inputs)
14
15 logits = outputs.logits
16 probabilities = torch.softmax(logits, dim=1).numpy()[0]
17 predicted_class = np.argmax(probabilities)
18
19 labels_map = {0: "Legitimate (Safe) ✅", 1: "FRAUD / PHISHING DETECTED 🚨"}
20
21 print(f"Analysis Verdict : {labels_map[predicted_class]}")
22 print(f"Confidence Matrix: Safe -> {probabilities[0]:.2%}, Fraud -> {probabilities[1]:.2%}")
23
24# Live test scenario
25scan_incoming_email("URGENT: Your online business portal access is frozen. Click here to confirm identity immediate.")