| 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 | | |
| |-------|------|----------|----------|----------------| | |
| ## Usage | |
| ### Python | |
| ```python | |
| from inference import CyberForgeInference | |
| inference = CyberForgeInference('./models') | |
| result = inference.predict('phishing_detection', features) | |
| ``` | |
| ### API | |
| ```python | |
| 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: | |
| ```bibtex | |
| @software{cyberforge2024, | |
| title = {CyberForge AI Security Models}, | |
| year = {2024}, | |
| url = {https://huggingface.co/Che237/cyberforge-models} | |
| } | |
| ``` | |