ATTACK BERT

🚀 ATT&CK BERT：一款網絡安全語言模型

ATT&CK BERT是一款基於sentence-transformers的網絡安全領域特定語言模型。它能將表示攻擊行為的句子映射到具有語義意義的嵌入向量上。語義相近的句子所對應的嵌入向量具有較高的餘弦相似度。

🚀 快速開始

ATT&CK BERT是一款基於sentence-transformers的網絡安全領域特定語言模型，它可以將描述攻擊行為的句子映射為具有語義意義的嵌入向量。語義相近的句子，其嵌入向量的餘弦相似度較高。

✨ 主要特性

• 領域針對性：專門針對網絡安全領域設計，能更好地處理該領域的文本。
• 語義映射：將攻擊行為的句子映射到有意義的嵌入向量，便於語義分析。
• 相似度計算：語義相近的句子嵌入向量餘弦相似度高，可用於相似度判斷。

📦 安裝指南

當你安裝了sentence-transformers後，使用該模型會變得很簡單：

pip install -U sentence-transformers

💻 使用示例

基礎用法

from sentence_transformers import SentenceTransformer
sentences = ["Attacker takes a screenshot", "Attacker captures the screen"]

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 BERT將文本映射到ATT&CK技術，請查看我們的工具SMET：https://github.com/basel-a/SMET

📄 許可證

本項目採用apache-2.0許可證。