IMMORTAL27105/phishing-email-detection
GitHub: IMMORTAL27105/phishing-email-detection
一个基于机器学习的钓鱼邮件检测原型系统,通过FastAPI后端配合Chrome扩展实现在Gmail中的实时钓鱼风险识别与提示。
Stars: 0 | Forks: 0
# 钓鱼邮件检测 – 安装与运行指南
本仓库包含一个**基于机器学习的钓鱼邮件检测原型**,包括:
* Python 后端 (FastAPI + ML 模型)
* 用于 Gmail 的 Chrome 扩展程序
请按照以下步骤在本地设置和运行项目。
## 1. 前置条件
请确保您的系统已安装以下软件:
* Python **3.8 或更高版本**
* Google Chrome 浏览器
* Git
## 2. 克隆仓库
```
git clone https://github.com/IMMORTAL27105/phishing-email-detection.git
cd phishing-email-detection
```
## 3. 后端设置 (Python)
### 3.1 创建虚拟环境 (推荐)
```
python -m venv venv
```
激活虚拟环境:
* **Windows**
venv\Scripts\activate
* **macOS / Linux**
source venv/bin/activate
### 3.2 安装依赖
```
cd backend
pip install -r requirements.txt
```
### 3.3 添加数据集
仓库中**不包含**数据集。
1. 下载钓鱼邮件数据集 (CSV 格式),此处我们使用了 kaggle 钓鱼邮件数据集和 enron 数据集。
2. 将其放置在:
backend/data/
示例:
```
backend/data/phishing_email.csv
```
### 3.4 训练模型
```
python train.py
```
这将生成:
```
backend/ml/model.pkl
backend/ml/vectorizer.pkl
```
### 3.5 运行后端服务器
```
uvicorn app:app --reload
```
保持此终端处于**运行状态**。
后端可通过以下地址访问:
```
http://127.0.0.1:8000
```
## 4. Chrome 扩展程序设置 (Gmail)
### 4.1 打开 Chrome 扩展程序页面
在 Chrome 中打开:
```
chrome://extensions/
```
启用右上角的**开发者模式**。
### 4.2 加载扩展程序
1. 点击**加载已解压的扩展程序**
2. 选择本仓库中的 `extension/` 文件夹
### 4.3 打开 Gmail
前往:
```
https://mail.google.com
```
该扩展程序将在 Gmail 上自动运行。
## 5. 如何验证是否正常工作
* 确保后端服务器正在运行
* 打开 Gmail
* 打开 Chrome DevTools → Console
* 您应该能看到来自 `content.js` 的日志
* 打开 Gmail 时,后端终端应会接收到请求
## 6. 注意事项
* 这是一个**学术原型**
* 该系统仅提供**风险指示**,而非绝对的分类
* 正常邮件偶尔可能会被误报
## 7. 常见问题
* **后端未运行** → 扩展程序将无法接收结果
* **缺少数据集** → 模型训练将失败
* **加载了错误的文件夹** → 扩展程序将无法运行
标签:AMSI绕过, Apex, AV绕过, Enron数据集, FastAPI, Gmail插件, Kaggle, Python, uvicorn, XAI, 前端后端分离, 可解释人工智能, 威胁检测, 学术原型, 异常检测, 数据可视化, 数据集训练, 文本分类, 无后门, 机器学习, 浏览器插件, 社会工程学防护, 网络安全, 网络钓鱼防御, 逆向工具, 邮件安全, 钓鱼邮件检测, 隐私保护