Instructions to use dev9269/ember-malware-rf with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Scikit-learn
How to use dev9269/ember-malware-rf with Scikit-learn:
from huggingface_hub import hf_hub_download import joblib model = joblib.load( hf_hub_download("dev9269/ember-malware-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 - Notebooks
- Google Colab
- Kaggle
File size: 950 Bytes
4a70bdd | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 | ---
license: mit
library_name: sklearn
tags:
- malware-detection
- ember
- cybersecurity
- random-forest
- pe-files
- security
---
# EMBER Malware Random Forest
A Random Forest classifier trained on the **EMBER** (Elastic Malware Benchmark for Empowering Researchers) dataset for PE malware detection.
## Model Details
- **Model Type**: Random Forest Classifier
- **Framework**: scikit-learn
- **Dataset**: EMBER v2 (1M+ PE samples)
- **Input**: 2381-dimension feature vector
- **Output**: Malicious / Benign classification
## Performance
Trained on the standard EMBER training split with optimized hyperparameters.
| Metric | Score |
|--------|-------|
| AUC-ROC | ~0.99 |
| Accuracy | ~97% |
| F1 Score | ~0.96 |
## Related
- [malware-detector-rf](https://huggingface.co/dev9269/malware-detector-rf) - Production variant
- [malware-detector-demo](https://huggingface.co/spaces/dev9269/malware-detector-demo) - Interactive demo
|