Views
No views yet
| Metric | Score |
|---|---|
| Accuracy | 91.0% |
| F1 | 90.2% |
| Precision | 83.1% |
| Recall | 98.6% |
| Train time | 7.6 min (P100 GPU) |
| Inference | 1022ms / 31 tokens-per-second |
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model = AutoModelForCausalLM.from_pretrained("Yusif-v/hephaestus-hdfs-0.5b")
4tokenizer = AutoTokenizer.from_pretrained("Yusif-v/hephaestus-0.5b")
5
6prompt = """<|im_start|>system
7You are a SOC analyst. Classify as NORMAL or ANOMALY.<|im_end|>
8<|im_start|>user
9powershell.exe -enc SQBFAFgAIAAoAE4AZQB3AC0ATwBiAGoAZQBjAHQ...<|im_end|>
10<|im_start|>assistant
11"""
12
13inputs = tokenizer(prompt, return_tensors="pt")
14outputs = model.generate(**inputs, max_new_tokens=64, do_sample=False)
15print(tokenizer.decode(outputs[0][inputs.input_ids.shape[1]:], skip_special_tokens=True))
16# Output: "ANOMALY. Suspicious encoded PowerShell command detected."