How to use Andou2yu/deep-attackg-bi-encoder with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("Andou2yu/deep-attackg-bi-encoder") sentences = [ "The ransomware injected malicious.dll into svchost.exe to execute its encryption routines under a trusted process.", "A rootkit modified UEFI runtime services to maintain control over the system.", "ZxxZ has relied on victims to open a malicious attachment delivered via email.", "FIN7 has used random junk code to obfuscate malware code." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4]
