Instructions to use ehsanaghaei/SecureBERT with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use ehsanaghaei/SecureBERT with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("fill-mask", model="ehsanaghaei/SecureBERT")# Load model directly from transformers import AutoTokenizer, AutoModelForMaskedLM tokenizer = AutoTokenizer.from_pretrained("ehsanaghaei/SecureBERT") model = AutoModelForMaskedLM.from_pretrained("ehsanaghaei/SecureBERT") - Inference
- Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -104,6 +104,7 @@ def predict_mask(sent, tokenizer, model, topk=10, print_results=True):
|
|
| 104 |
|
| 105 |
|
| 106 |
# Reference
|
|
|
|
| 107 |
@inproceedings{aghaei2023securebert,
|
| 108 |
title={SecureBERT: A Domain-Specific Language Model for Cybersecurity},
|
| 109 |
author={Aghaei, Ehsan and Niu, Xi and Shadid, Waseem and Al-Shaer, Ehab},
|
|
@@ -113,3 +114,4 @@ pages={39--56},
|
|
| 113 |
year={2023},
|
| 114 |
organization={Springer}
|
| 115 |
}
|
|
|
|
|
|
| 104 |
|
| 105 |
|
| 106 |
# Reference
|
| 107 |
+
```
|
| 108 |
@inproceedings{aghaei2023securebert,
|
| 109 |
title={SecureBERT: A Domain-Specific Language Model for Cybersecurity},
|
| 110 |
author={Aghaei, Ehsan and Niu, Xi and Shadid, Waseem and Al-Shaer, Ehab},
|
|
|
|
| 114 |
year={2023},
|
| 115 |
organization={Springer}
|
| 116 |
}
|
| 117 |
+
```
|