metadata
license: mit
tags:
- cybersecurity
- threat-detection
- phishing
- malware
- security
language:
- en
CyberForge AI Models
Production-ready machine learning models for cybersecurity threat detection.
Models Included
| Model | Task | Accuracy | F1 Score | Inference Time |
|---|---|---|---|---|
| phishing_detection | random_forest | 0.9890 | 0.9890 | 0.02ms |
| malware_detection | gradient_boosting | 0.9985 | 0.9985 | 0.00ms |
| anomaly_detection | random_forest | 0.9990 | 0.9990 | 0.01ms |
| web_attack_detection | random_forest | 1.0000 | 1.0000 | 0.04ms |
Usage
Python
from inference import CyberForgeInference
inference = CyberForgeInference('./models')
result = inference.predict('phishing_detection', features)
API
import requests
response = requests.post(
'https://huggingface.co/spaces/Che237/cyberforge/predict',
json={'model_name': 'phishing_detection', 'features': features}
)
Model Details
- Framework: scikit-learn
- Python Version: 3.11+
- License: MIT
Citation
If you use these models, please cite:
@software{cyberforge2024,
title = {CyberForge AI Security Models},
year = {2024},
url = {https://huggingface.co/Che237/cyberforge-models}
}