使用 FastAPI 和机器学习构建实时信用卡欺诈检测系统

王林
发布: 2024-08-13 06:54:33
原创
620 人浏览过

Building a Real-Time Credit Card Fraud Detection System with FastAPI and Machine Learning

介绍

信用卡欺诈对金融业构成重大威胁,每年造成数十亿美元的损失。为了解决这个问题,人们开发了机器学习模型来实时检测和防止欺诈交易。在本文中,我们将逐步介绍使用 FastAPI(Python 的现代 Web 框架)以及在 Kaggle 流行的信用卡欺诈检测数据集上训练的随机森林分类器构建实时信用卡欺诈检测系统的过程。

项目概况

该项目的目标是创建一个 Web 服务来预测信用卡交易欺诈的可能性。该服务接受交易数据,对其进行预处理,然后返回预测以及欺诈概率。该系统设计快速、可扩展,并且易于集成到现有的金融系统中。

关键部件

  1. 机器学习模型:经过训练以区分欺诈交易和合法交易的随机森林分类器。
  2. 数据预处理:交易特征标准化,确保模型达到最佳性能。
  3. API:使用 FastAPI 构建的 RESTful API,用于实时处理预测请求。

第 1 步:准备数据集

本项目使用的数据集是来自 Kaggle 的信用卡欺诈检测数据集,其中包含 284,807 笔交易,其中只有 492 笔是欺诈交易。这种类别不平衡带来了挑战,但可以通过对少数类别进行过采样来解决。

数据预处理

这些功能首先使用 scikit-learn 的 StandardScaler 进行标准化。然后将数据集分为训练集和测试集。鉴于不平衡,在训练模型之前应用 RandomOverSampler 技术来平衡类别。

from sklearn.preprocessing import StandardScaler
from imblearn.over_sampling import RandomOverSampler

# Standardize features
scaler = StandardScaler()
X_scaled = scaler.fit_transform(X)

# Balance the dataset
ros = RandomOverSampler(random_state=42)
X_resampled, y_resampled = ros.fit_resample(X_scaled, y)
登录后复制

第 2 步:训练机器学习模型

我们训练了一个随机森林分类器,它非常适合处理不平衡的数据集并提供可靠的预测。该模型在过采样数据上进行训练,并使用准确度、精确度、召回率和 AUC-ROC 曲线来评估其性能。

from sklearn.ensemble import RandomForestClassifier
from sklearn.metrics import classification_report, roc_auc_score

# Train the model
model = RandomForestClassifier(n_estimators=100, random_state=42)
model.fit(X_resampled, y_resampled)

# Evaluate the model
y_pred = model.predict(X_test_scaled)
print(classification_report(y_test, y_pred))
print("AUC-ROC:", roc_auc_score(y_test, model.predict_proba(X_test_scaled)[:, 1]))
登录后复制

第 3 步:构建 FastAPI 应用程序

使用 joblib 保存训练好的模型和缩放器后,我们继续构建 FastAPI 应用程序。选择 FastAPI 是因为其速度快且易于使用,使其成为实时应用程序的理想选择。

创建 API

FastAPI 应用程序定义了一个 POST 端点 /predict/,它接受交易数据、对其进行处理并返回模型的预测和概率。

from fastapi import FastAPI, HTTPException
from pydantic import BaseModel
import joblib
import pandas as pd

# Load the trained model and scaler
model = joblib.load("random_forest_model.pkl")
scaler = joblib.load("scaler.pkl")

app = FastAPI()

class Transaction(BaseModel):
    V1: float
    V2: float
    # Include all other features used in your model
    Amount: float

@app.post("/predict/")
def predict(transaction: Transaction):
    try:
        data = pd.DataFrame([transaction.dict()])
        scaled_data = scaler.transform(data)
        prediction = model.predict(scaled_data)
        prediction_proba = model.predict_proba(scaled_data)
        return {"fraud_prediction": int(prediction[0]), "probability": float(prediction_proba[0][1])}
    except Exception as e:
        raise HTTPException(status_code=400, detail=str(e))
登录后复制
第 4 步:部署应用程序

要在本地测试应用程序,您可以使用 uvicorn 运行 FastAPI 服务器并向 /predict/ 端点发送 POST 请求。该服务将处理传入请求、扩展数据并返回交易是否存在欺诈。

本地运行 API

uvicorn main:app --reload
登录后复制
然后您可以使用curl 或Postman 等工具测试API:


curl -X POST http://127.0.0.1:8000/predict/ \
-H "Content-Type: application/json" \
-d '{"V1": -1.359807134, "V2": -0.072781173, ..., "Amount": 149.62}'
登录后复制
API 将返回一个 JSON 对象,其中包含欺诈预测和相关概率。

结论

在本文中,我们构建了一个实时信用卡欺诈检测系统,该系统将机器学习与现代 Web 框架相结合。 github 链接在这里。该系统旨在处理实时交易数据并提供即时预测,使其成为金融机构打击欺诈的宝贵工具。

通过使用 FastAPI 部署此模型,我们确保服务不仅快速而且可扩展,能够同时处理多个请求。该项目可以通过更复杂的模型、改进的特征工程或与生产环境的集成来进一步扩展。

下一步

要进一步增强系统,请考虑以下事项:

  1. 模型改进:尝试更高级的模型,如 XGBoost 或神经网络。
  2. 特征工程:探索可能提高模型准确性的其他功能。
  3. 实际部署:在 AWS 或 GCP 等云平台上部署应用程序以供生产使用。

以上是使用 FastAPI 和机器学习构建实时信用卡欺诈检测系统的详细内容。更多信息请关注PHP中文网其他相关文章!

来源:dev.to
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板
关于我们 免责声明 Sitemap
PHP中文网:公益在线PHP培训,帮助PHP学习者快速成长!