pipeline_tag: 句子相似度
tags:

• 网络安全
• 句子嵌入
• 句子相似度

ATT&CK BERT：网络安全专用语言模型

ATT&CK BERT是基于sentence-transformers的网络安全领域专用语言模型。
该模型可将描述攻击行为的句子映射为具有语义意义的嵌入向量，语义相近的句子其嵌入向量的余弦相似度较高。

使用方法（Sentence-Transformers）

安装sentence-transformers后即可轻松调用本模型：

pip install -U sentence-transformers

调用示例：

from sentence_transformers import SentenceTransformer
sentences = ["攻击者截取屏幕截图", "攻击者捕获屏幕内容"]

model = SentenceTransformer('basel/ATTACK-BERT')
embeddings = model.encode(sentences)

from sklearn.metrics.pairwise import cosine_similarity
print(cosine_similarity([embeddings[0]], [embeddings[1]]))

若需将文本映射到ATT&CK攻击技术，请使用我们的工具SMET：
https://github.com/basel-a/SMET

许可证：
apache-2.0