Update README.md
Browse files
README.md
CHANGED
|
@@ -9,6 +9,8 @@ datasets:
|
|
| 9 |
<!-- Provide a quick summary of what the model is/does. -->
|
| 10 |
A meta-llama/Meta-Llama-Guard-2-8B model fine-tuned on the nvidia/Aegis-AI-Content-Safety-Dataset-1.0 dataset. A total of 3099 examples are in the training set.
|
| 11 |
|
|
|
|
|
|
|
| 12 |
This is a multi-label text classifier that has 14 categories:
|
| 13 |
- "0": "Controlled/Regulated Substances"
|
| 14 |
- "1": "Criminal Planning/Confessions"
|
|
|
|
| 9 |
<!-- Provide a quick summary of what the model is/does. -->
|
| 10 |
A meta-llama/Meta-Llama-Guard-2-8B model fine-tuned on the nvidia/Aegis-AI-Content-Safety-Dataset-1.0 dataset. A total of 3099 examples are in the training set.
|
| 11 |
|
| 12 |
+
The model was finetuned using huggingface Trainer class, with `1500` max_steps.
|
| 13 |
+
|
| 14 |
This is a multi-label text classifier that has 14 categories:
|
| 15 |
- "0": "Controlled/Regulated Substances"
|
| 16 |
- "1": "Criminal Planning/Confessions"
|