Views
No views yet
Qwen3ForSequenceClassification (Qwen3-4B, 36 couches, hidden 2560, bfloat16).Qwen/Qwen3-4B (Apache-2.0, multilingue FR/EN).| Index | Signification |
|---|---|
0 | Non Green IT |
1 | Green IT |
1import torch
2from transformers import AutoModelForSequenceClassification, AutoTokenizer
3
4REPO = "KaRn1zC/greentech-greenit-classifier"
5tokenizer = AutoTokenizer.from_pretrained(REPO)
6model = AutoModelForSequenceClassification.from_pretrained(REPO, dtype="auto").eval()
7
8# Calibration persistée dans le dépôt (temperature.json / optimal_threshold.json)
9TEMPERATURE = 1.3949
10THRESHOLD = 0.155 # seuil sur la probabilité de la classe Green IT (index 1)
11
12texte = "Titre de l'article. Résumé dense de son contenu..."
13inputs = tokenizer(texte, truncation=True, max_length=512, return_tensors="pt")
14with torch.no_grad():
15 logits = model(**inputs).logits
16
17probs = torch.softmax(logits / TEMPERATURE, dim=-1)
18p_green = probs[0, 1].item()
19est_green_it = p_green >= THRESHOLD
20print(f"Green IT : {est_green_it} (probabilité = {p_green:.3f})")La calibration (temperature scaling T = 1,3949 + seuil de décision 0,155) est appliquée après lesoftmax. Sans elle, on retombe sur le seuil naïf de 0,5, moins performant sur cette distribution déséquilibrée (~18 % de Green IT).
Helsinki-NLP/opus-mt), variantes exclues des
ensembles de validation/test pour éviter toute fuite.q,k,v,o,gate,up,down_proj), r=16, alpha=32, dropout=0,05, rsLoRA.lr=1e-4 (cosine, warmup 0,06), max_length=512, bfloat16,
gradient checkpointing, pondération de classe [1.0, 10.5] (perte weighted cross-entropy).MultilabelStratifiedKFold (langue × label), K=3 folds × 2 seeds (6 entraînements).density=0,5), tête de classification moyennée.| Métrique | Valeur |
|---|---|
| MCC (Matthews) | 0,6238 ± 0,0103 |
| F1 | 0,6861 |
| Rappel (Green IT) | 0,8913 |
| Précision (Green IT) | 0,5573 |
| Latence moyenne (RX 7900 XTX, ROCm) | ~58 ms / article |
uv run python scripts/retrain_pipeline.py train-cv --model=qwen3KaRn1zC. GreenTech Intelligence — Classifieur Green IT (Qwen3-4B + LoRA TIES), 2026. Modèle de base : Qwen3-4B (Apache-2.0).