Update README.md
Browse files
README.md
CHANGED
|
@@ -27,16 +27,20 @@ This model is designed to allow you to predict a single CWE given your descripti
|
|
| 27 |
### Model Description
|
| 28 |
|
| 29 |
<!-- Provide a longer summary of what this model is. -->
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 30 |
|
| 31 |
This is the model card of a 🤗 transformers model that has been pushed on the Hub. This model card has been automatically generated.
|
| 32 |
|
| 33 |
-
- **Developed by:** [
|
| 34 |
-
- **
|
| 35 |
-
- **
|
| 36 |
-
- **
|
| 37 |
-
- **
|
| 38 |
-
- **License:** [More Information Needed]
|
| 39 |
-
- **Finetuned from model [optional]:** [More Information Needed]
|
| 40 |
|
| 41 |
### Model Sources [optional]
|
| 42 |
|
|
|
|
| 27 |
### Model Description
|
| 28 |
|
| 29 |
<!-- Provide a longer summary of what this model is. -->
|
| 30 |
+
The model takes in text and predicts a single CWE. On it's last training run it saw the following:
|
| 31 |
+
|
| 32 |
+
- Training Loss: 1.158700
|
| 33 |
+
- Validation Loss: 1.199677
|
| 34 |
+
- Accuracy: 0.71136
|
| 35 |
+
- F1: 0.229855
|
| 36 |
|
| 37 |
This is the model card of a 🤗 transformers model that has been pushed on the Hub. This model card has been automatically generated.
|
| 38 |
|
| 39 |
+
- **Developed by:** [mulliken](https://huggingface.co/mulliken)
|
| 40 |
+
- **Model type:** BERT
|
| 41 |
+
- **Language(s) (NLP):** English
|
| 42 |
+
- **License:** Apache 2.0
|
| 43 |
+
- **Finetuned from model:** [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased)
|
|
|
|
|
|
|
| 44 |
|
| 45 |
### Model Sources [optional]
|
| 46 |
|