Malware Detector - Random Forest
A Random Forest classifier for PE (Portable Executable) malware detection, trained on the EMBER dataset.
Model Details
- Model Type: Random Forest Classifier (scikit-learn)
- Training Data: EMBER dataset (features from PE file headers, sections, imports, exports, etc.)
- Framework: scikit-learn (LightGBM backend)
- Input: Extracted PE feature vector (2381 dimensions)
- Output: Malicious / Benign classification with confidence score
Usage
import joblib
import numpy as np
model = joblib.load("model.pkl")
features = np.load("sample_features.npy") # 2381-dim feature vector
prediction = model.predict([features])[0]
confidence = model.predict_proba([features])[0]
print(f"Malicious: {bool(prediction)}")
print(f"Confidence: {max(confidence):.2%}")
Performance
| Metric | Score |
|---|---|
| Accuracy | ~96% |
| Precision | ~0.95 |
| Recall | ~0.94 |
| F1 Score | ~0.94 |
Related Models
- ember-malware-rf - EMBER-trained variant
- malware-detector-demo - Interactive demo Space
- Downloads last month
- -
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support
from huggingface_hub import hf_hub_download import joblib model = joblib.load( hf_hub_download("dev9269/malware-detector-rf", "sklearn_model.joblib") ) # only load pickle files from sources you trust # read more about it here https://skops.readthedocs.io/en/stable/persistence.html