Skip to content

Commit b3e874a

Browse files
chg: Release 3.1.0
Source field in datasets, dynamic multi-source dataset card, per-class metrics and best checkpoint selection for severity trainer, card templates moved to vulntrain/cards/. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
1 parent 7ced65f commit b3e874a

2 files changed

Lines changed: 21 additions & 1 deletion

File tree

CHANGELOG.md

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,25 @@
11
# Changelog
22

3+
## Release 3.1.0 (2026-04-06)
4+
5+
### What's New
6+
7+
#### Datasets
8+
9+
- **Source field**: each vulnerability entry now includes a `source` field identifying its origin (cvelistv5, github, pysec, cnvd, csaf_*).
10+
- **Dynamic dataset card for multi-source datasets**: when generating a dataset from multiple sources (e.g., `--sources cvelistv5,github,csaf_redhat,csaf_cisco,csaf_cisa,pysec`), a dataset card is now automatically generated with a per-source breakdown table showing entry counts and percentages.
11+
12+
#### Training
13+
14+
- **Per-class metrics for severity trainer** (`classify_severity.py`): `compute_metrics` now reports precision, recall, and F1 per class (Low/Medium/High/Critical) alongside overall accuracy and macro F1.
15+
- **Best model checkpoint selection** (`classify_severity.py`): model selected by accuracy instead of eval_loss, `save_total_limit` increased from 2 to 3.
16+
17+
### Changes
18+
19+
- Moved all HuggingFace card templates (dataset cards, model cards) to a dedicated `vulntrain/cards/` directory.
20+
- Updated dependencies.
21+
22+
323
## Release 3.0.0 (2026-04-03)
424

525
### What's New

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ build-backend = "poetry.core.masonry.api"
55

66
[project]
77
name = "VulnTrain"
8-
version = "3.0.0"
8+
version = "3.1.0"
99
description = "Generate datasets amd models based on vulnerabilities data from Vulnerability-Lookup."
1010
authors = [
1111
{name = "Cédric Bonhomme", email = "cedric.bonhomme@circl.lu"},

0 commit comments

Comments
 (0)