Skip to content

Commit c94d3d0

Browse files
chg: [RELEASE] Bumped release number and updated CHANGELOG.
1 parent e8e62bc commit c94d3d0

File tree

3 files changed

+30
-3
lines changed

3 files changed

+30
-3
lines changed

CHANGELOG.md

+9
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,14 @@
11
# Changelog
22

3+
## Release 1.1.0 (2025-02-27)
4+
5+
### News
6+
7+
- Trainers: Support of roberta-base for the text classifier with improved
8+
settings for TrainingArguments.
9+
- Validators: Validator for severity classification.
10+
11+
312
## Release 1.0.0 (2025-02-25)
413

514
### News

README.md

+20-2
Original file line numberDiff line numberDiff line change
@@ -109,10 +109,28 @@ Using CUDA (Nvidia GPU).
109109

110110
#### Training for classification
111111

112+
```bash
113+
$ vulntrain-train-severity-classification --help
114+
usage: vulntrain-train-severity-classification [-h] [--base-model {distilbert-base-uncased,roberta-base}] [--dataset-id DATASET_ID] --repo-id REPO_ID [--model-save-dir MODEL_SAVE_DIR]
115+
116+
Train a vulnerability classification model with a mapping on the severity.
117+
118+
options:
119+
-h, --help show this help message and exit
120+
--base-model {distilbert-base-uncased,roberta-base}
121+
Base model to use.
122+
--dataset-id DATASET_ID
123+
Path of the dataset. Local dataset or repository on the HF hub.
124+
--repo-id REPO_ID The name of the repository you want to push your object to. It should contain your organization name when pushing to a given organization.
125+
--model-save-dir MODEL_SAVE_DIR
126+
The path to a directory where the tokenizer and the model will be saved.
112127

113-
- distilbert with CVS scores mapping
114-
- tf-idf on the vulnerability descriptions.
115128

129+
$ vulntrain-train-severity-classification --base-model roberta-base --dataset-id CIRCL/vulnerability-scores --repo-id CIRCL/vulnerability-severity-classification-roberta-base
130+
...
131+
...
132+
...
133+
```
116134

117135

118136
### Validation

pyproject.toml

+1-1
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 = "1.0.0"
8+
version = "1.1.0"
99
description = "Generate datasets amd models based on vulnerabilities data from Vulnerability-Lookup."
1010
authors = [
1111
{name = "Cédric Bonhomme",email = "[email protected]"}

0 commit comments

Comments
 (0)