README.md

---
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

```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}
}
```