Sentence Similarity
Transformers
PyTorch
mpnet
feature-extraction
cybersecurity
sentence-embedding
text-embeddings-inference
Instructions to use basel/ATTACK-BERT with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use basel/ATTACK-BERT with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("basel/ATTACK-BERT") model = AutoModel.from_pretrained("basel/ATTACK-BERT") - Inference
- Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -37,3 +37,4 @@ from sklearn.metrics.pairwise import cosine_similarity
|
|
| 37 |
print(cosine_similarity([embeddings[0]], [embeddings[1]]))
|
| 38 |
```
|
| 39 |
|
|
|
|
|
|
| 37 |
print(cosine_similarity([embeddings[0]], [embeddings[1]]))
|
| 38 |
```
|
| 39 |
|
| 40 |
+
To use ATT&CK BERT to map text to ATT&CK techniques Check our tool SMET: https://github.com/basel-a/SMET
|