mulliken commited on
Commit
544e014
·
verified ·
1 Parent(s): c646f3f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +11 -7
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:** [More Information Needed]
34
- - **Funded by [optional]:** [More Information Needed]
35
- - **Shared by [optional]:** [More Information Needed]
36
- - **Model type:** [More Information Needed]
37
- - **Language(s) (NLP):** [More Information Needed]
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