【机器学习】AUC、F1分数不再迷茫:图解Scikit-Learn模型评估与选择核心技巧

Langchain系列文章目录

01-玩转LangChain:从模型调用到Prompt模板与输出解析的完整指南
02-玩转 LangChain Memory 模块:四种记忆类型详解及应用场景全覆盖
03-全面掌握 LangChain:从核心链条构建到动态任务分配的实战指南
04-玩转 LangChain:从文档加载到高效问答系统构建的全程实战
05-玩转 LangChain:深度评估问答系统的三种高效方法(示例生成、手动评估与LLM辅助评估)
06-从 0 到 1 掌握 LangChain Agents:自定义工具 + LLM 打造智能工作流!
07-【深度解析】从GPT-1到GPT-4:ChatGPT背后的核心原理全揭秘

PyTorch系列文章目录

Python系列文章目录

机器学习系列文章目录

01-什么是机器学习?从零基础到自动驾驶案例全解析
02-从过拟合到强化学习:机器学习核心知识全解析
03-从零精通机器学习:线性回归入门
04-逻辑回归 vs. 线性回归:一文搞懂两者的区别与应用
05-决策树算法全解析:从零基础到Titanic实战,一文搞定机器学习经典模型
06-集成学习与随机森林:从理论到实践的全面解析
07-支持向量机(SVM):从入门到精通的机器学习利器
08-【机器学习】KNN算法入门:从零到电影推荐实战
09-【机器学习】朴素贝叶斯入门:从零到垃圾邮件过滤实战
10-【机器学习】聚类算法全解析:K-Means、层次聚类、DBSCAN在市场细分的应用
11-【机器学习】降维与特征选择全攻略:PCA、LDA与特征选择方法详解
12-【机器学习】手把手教你构建神经网络:从零到手写数字识别实战
13-【机器学习】从零开始学习卷积神经网络(CNN):原理、架构与应用
14-【机器学习】RNN与LSTM全攻略:解锁序列数据的秘密
15-【机器学习】GAN从入门到实战:手把手教你实现生成对抗网络
16-【机器学习】强化学习入门:从零掌握 Agent 到 DQN 核心概念与 Gym 实战
17-【机器学习】AUC、F1分数不再迷茫:图解Scikit-Learn模型评估与选择核心技巧


前言

大家好!欢迎来到我们机器学习系列文章的第 17 天。在前几天的学习中,我们已经接触了多种机器学习模型,如线性回归、逻辑回归、决策树、随机森林等。然而,构建模型只是第一步,更关键的问题是:我们如何知道哪个模型更好?如何科学地评估模型的性能,并从中选择最适合我们任务的模型呢? 这就是我们今天要深入探讨的主题——模型评估与选择。选择错误的评估指标或方法,可能会让你对模型的“性能”产生误判,尤其是在处理现实世界中常见的不平衡数据时。本文将系统梳理分类和回归任务中常用的评估指标,详解交叉验证技术,并结合 Scikit-Learn 代码实例,带你掌握科学评估和选择模型的必备技能,避开常见的“陷阱”。

一、为什么需要模型评估?

想象一下,你训练了两个不同的模型来预测客户是否会流失。模型 A 在训练集上表现完美,模型 B 稍逊一筹。我们能直接说模型 A 更好吗?未必!模型可能只是“记住”了训练数据(过拟合),在新的、未见过的数据上表现会很差。

模型评估的核心目的在于:

  1. 量化模型性能:提供客观、可比较的指标来衡量模型在特定任务上的表现好坏。
  2. 模型比较与选择:在多个候选模型(不同的算法或同一算法不同超参数)之间进行选择。
  3. 诊断模型问题:帮助我们理解模型的强项和弱点(例如,在哪类样本上容易出错),指导模型优化方向。
  4. 泛化能力评估:评估模型在未见过的数据上的表现能力,这才是模型真正投入使用时我们关心的。

因此,科学的模型评估是连接模型训练和实际应用的关键桥梁。

二、分类模型评估指标

分类任务是最常见的机器学习任务之一(如判断邮件是否为垃圾邮件、图片属于哪个类别、用户是否会点击广告等)。评估分类模型性能的指标众多,适用于不同的场景。

2.1 混淆矩阵 (Confusion Matrix)

混淆矩阵是理解分类模型性能的基础,它以矩阵形式总结了模型预测结果与真实标签的对比情况。对于二分类问题,混淆矩阵通常表示为:

预测为正类 (Positive)预测为负类 (Negative)
实际为正类真阳性 (TP)假阴性 (FN)
实际为负类假阳性 (FP)真阴性 (TN)
  • 真阳性 (True Positive, TP): 模型预测为正类,实际也为正类。
  • 假阴性 (False Negative, FN): 模型预测为负类,实际却为正类。(模型漏报了正类)
  • 假阳性 (False Positive, FP): 模型预测为正类,实际却为负类。(模型误报了正类,也称 Type I error)
  • 真阴性 (True Negative, TN): 模型预测为负类,实际也为负类。

混淆矩阵本身不直接给出单一的评估分数,但它是计算许多其他重要指标的基础。

# Scikit-Learn 示例: 计算混淆矩阵
from sklearn.metrics import confusion_matrix
from sklearn.linear_model import LogisticRegression
from sklearn.model_selection import train_test_split
from sklearn.datasets import make_classification

# 生成模拟数据
X, y = make_classification(n_samples=1000, n_features=20, random_state=42)
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=42)

# 训练一个简单模型
model = LogisticRegression()
model.fit(X_train, y_train)
y_pred = model.predict(X_test)

# 计算混淆矩阵
cm = confusion_matrix(y_test, y_pred)
print("混淆矩阵:")
print(cm)
# 输出示例:
# 混淆矩阵:
# [[124  20]
#  [ 29 127]]
# 这意味着: TN=124, FP=20, FN=29, TP=127

2.2 准确率 (Accuracy)

准确率是最直观的评估指标,表示模型预测正确的样本数占总样本数的比例。

Accuracy = T P + T N T P + T N + F P + F N = 预测正确的样本数 总样本数 \text{Accuracy} = \frac{TP + TN}{TP + TN + FP + FN} = \frac{\text{预测正确的样本数}}{\text{总样本数}} Accuracy=TP+TN+FP+FNTP+TN=总样本数预测正确的样本数

优点:简单易懂,能宏观反映模型的整体性能。
缺点:在数据不平衡时具有极大的误导性!

2.2.1 准确率陷阱:不平衡数据集

场景:假设我们有一个预测用户是否患有某种罕见疾病的数据集,其中 99% 的用户是健康的(负类),只有 1% 的用户患病(正类)。

如果一个模型非常“懒惰”,它总是预测所有人都是健康的(即总是预测负类),那么:

  • TP = 0 (没有病人被正确预测)
  • FN = 10 (假设有 1000 个样本,10 个病人,都被漏报)
  • FP = 0 (没有健康人被误报)
  • TN = 990 (990 个健康人被正确预测)

此时,该模型的准确率为:
Accuracy = 0 + 990 0 + 990 + 0 + 10 = 990 1000 = 99 % \text{Accuracy} = \frac{0 + 990}{0 + 990 + 0 + 10} = \frac{990}{1000} = 99\% Accuracy=0+990+0+100+990=1000990=99%
这个准确率看起来非常高,但模型完全没有识别出任何一个病人,对于疾病检测这个任务来说,这样的模型是毫无价值的!这就是准确率在不平衡数据集上的“陷阱”。

# Scikit-Learn 示例: 计算准确率
from sklearn.metrics import accuracy_score

accuracy = accuracy_score(y_test, y_pred)
print(f"准确率: {accuracy:.4f}")
# 输出示例:
# 准确率: 0.8367

2.3 精确率 (Precision)

精确率(也叫查准率)关注的是模型预测为正类的样本中,有多少是真正的正类

Precision = T P T P + F P = 预测为正且实际为正的样本数 所有预测为正的样本数 \text{Precision} = \frac{TP}{TP + FP} = \frac{\text{预测为正且实际为正的样本数}}{\text{所有预测为正的样本数}} Precision=TP+FPTP=所有预测为正的样本数预测为正且实际为正的样本数

应用场景:当我们希望**尽量避免误报(FP)**时,精确率很重要。例如:

  • 垃圾邮件过滤:我们不希望把重要的邮件误判为垃圾邮件(FP),宁可放过一些垃圾邮件(FN)。
  • 股票预测:预测某支股票会上涨,如果预测错了(FP),可能会导致亏损,因此希望预测上涨的判断尽可能准。

2.4 召回率 (Recall)

召回率(也叫查全率、敏感度 Sensitivity)关注的是实际为正类的样本中,有多少被模型成功预测出来了

Recall = T P T P + F N = 预测为正且实际为正的样本数 所有实际为正的样本数 \text{Recall} = \frac{TP}{TP + FN} = \frac{\text{预测为正且实际为正的样本数}}{\text{所有实际为正的样本数}} Recall=TP+FNTP=所有实际为正的样本数预测为正且实际为正的样本数

应用场景:当我们希望**尽量避免漏报(FN)**时,召回率很重要。例如:

  • 疾病诊断:我们不希望漏掉任何一个真正的病人(FN),宁可误诊一些健康人(FP),因为漏诊的后果可能很严重。
  • 金融欺诈检测:希望尽可能多地找出欺诈交易(FN),即使可能误判一些正常交易(FP)。

2.5 F1 分数 (F1 Score)

精确率和召回率往往是相互矛盾的(trade-off)。提高精确率可能会降低召回率,反之亦然。F1 分数是精确率和召回率的调和平均数,旨在综合这两者。

F 1 = 2 × Precision × Recall Precision + Recall = 2 T P 2 T P + F P + F N F_1 = 2 \times \frac{\text{Precision} \times \text{Recall}}{\text{Precision} + \text{Recall}} = \frac{2TP}{2TP + FP + FN} F1=2×Precision+RecallPrecision×Recall=2TP+FP+FN2TP

特点

  • F1 分数的值域为 [0, 1],越高越好。
  • 当精确率和召回率都较高时,F1 分数也会较高。
  • 相比算术平均,调和平均更看重较小的值,因此只有当 Precision 和 Recall 都比较高时,F1 才会高。

应用场景:当精确率和召回率同等重要时,或者在不平衡数据下,F1 分数是比准确率更可靠的指标。

# Scikit-Learn 示例: 计算精确率、召回率、F1分数
from sklearn.metrics import precision_score, recall_score, f1_score

precision = precision_score(y_test, y_pred)
recall = recall_score(y_test, y_pred)
f1 = f1_score(y_test, y_pred)

print(f"精确率 (Precision): {precision:.4f}")
print(f"召回率 (Recall): {recall:.4f}")
print(f"F1 分数 (F1 Score): {f1:.4f}")

# 同时输出多个指标
from sklearn.metrics import classification_report
print("\n分类报告:")
print(classification_report(y_test, y_pred))

输出示例:

精确率 (Precision): 0.8639
召回率 (Recall): 0.8141
F1 分数 (F1 Score): 0.8383

分类报告:
              precision    recall  f1-score   support

           0       0.81      0.86      0.83       144
           1       0.86      0.81      0.84       156

    accuracy                           0.84       300
   macro avg       0.84      0.84      0.84       300
weighted avg       0.84      0.84      0.84       300

classification_report 提供了每个类别的 Precision, Recall, F1-score 以及它们的宏平均(macro avg)和加权平均(weighted avg),还有整体的准确率。

三、ROC 曲线与 AUC 值详解

3.1 ROC 曲线 (Receiver Operating Characteristic Curve)

很多分类模型(如逻辑回归、SVM)输出的不是直接的类别标签(0 或 1),而是一个概率值置信度分数,表示样本属于正类的可能性。我们需要设定一个阈值 (Threshold),当概率大于阈值时预测为正类,否则预测为负类。

阈值的选择会直接影响 TP, FP, FN, TN 的值,进而影响精确率和召回率。ROC 曲线正是为了可视化不同阈值下模型性能变化而生的。

  • 横坐标 (X-axis)假阳性率 (False Positive Rate, FPR),也叫 1 - Specificity。
    F P R = F P F P + T N = 预测为正但实际为负的样本数 所有实际为负的样本数 FPR = \frac{FP}{FP + TN} = \frac{\text{预测为正但实际为负的样本数}}{\text{所有实际为负的样本数}} FPR=FP+TNFP=所有实际为负的样本数预测为正但实际为负的样本数
    它表示所有负类样本中,被模型错误预测为正类的比例。我们希望 FPR 尽可能小。
  • 纵坐标 (Y-axis)真阳性率 (True Positive Rate, TPR),也就是召回率 (Recall)
    T P R = T P T P + F N = Recall TPR = \frac{TP}{TP + FN} = \text{Recall} TPR=TP+FNTP=Recall
    它表示所有正类样本中,被模型正确预测为正类的比例。我们希望 TPR 尽可能大。

如何绘制 ROC 曲线?

  1. 模型对测试集所有样本进行预测,得到每个样本属于正类的概率分数。
  2. 将这些概率分数从高到低排序。
  3. 以每个概率分数作为潜在的阈值:
    • 将阈值设为略大于最高概率,此时所有样本都被预测为负类,TP=0, FP=0,所以 TPR=0, FPR=0。这是曲线的起点 (0, 0)。
    • 逐渐降低阈值,每次遇到一个实际为正类的样本,TP 增加,TPR 增加(曲线向上移动);每次遇到一个实际为负类的样本,FP 增加,FPR 增加(曲线向右移动)。
    • 将阈值设为略小于最低概率,此时所有样本都被预测为正类,TN=0, FN=0,所以 TPR=1, FPR=1。这是曲线的终点 (1, 1)。
  4. 将所有 (FPR, TPR) 点连接起来,就形成了 ROC 曲线。

解读 ROC 曲线

  • 曲线越靠近左上角 (0, 1) 点,表示模型性能越好(在较低的 FPR 下获得较高的 TPR)。
  • 对角线(从 (0, 0) 到 (1, 1))表示随机猜测的模型性能。
  • 曲线完全在对角线上方,表示模型优于随机猜测。
  • 曲线在对角线下方,表示模型比随机猜测还差(可能需要反转预测)。

3.2 AUC (Area Under the ROC Curve) 值

AUC 值就是 ROC 曲线下的面积

  • AUC 的值域为 [0, 1]。
  • AUC = 1:完美分类器。
  • AUC = 0.5:随机猜测。
  • AUC < 0.5:比随机猜测还差。
  • 通常 AUC 值越大,表示模型区分正负样本的能力越强。

AUC 的优点

  • 不依赖于特定的阈值:它衡量的是模型在所有可能阈值下的整体性能。
  • 对样本类别分布不敏感:相比准确率,AUC 在不平衡数据集上是更稳健的评估指标。即使正负样本比例失衡,AUC 仍能较好地反映模型的排序能力。

AUC 的直观理解:AUC 值可以看作是随机抽取一个正样本和一个负样本,模型将正样本预测为正类的概率大于将负样本预测为正类的概率的可能性

# Scikit-Learn 示例: 计算 ROC 曲线和 AUC 值
from sklearn.metrics import roc_curve, roc_auc_score
import matplotlib.pyplot as plt

# 获取模型预测概率 (需要用 predict_proba)
y_pred_proba = model.predict_proba(X_test)[:, 1] # 获取正类的概率

# 计算 ROC 曲线的 FPR, TPR 和 阈值
fpr, tpr, thresholds = roc_curve(y_test, y_pred_proba)

# 计算 AUC 值
auc_value = roc_auc_score(y_test, y_pred_proba)
print(f"AUC 值: {auc_value:.4f}")

# 绘制 ROC 曲线
plt.figure(figsize=(8, 6))
plt.plot(fpr, tpr, color='darkorange', lw=2, label=f'ROC curve (area = {auc_value:.2f})')
plt.plot([0, 1], [0, 1], color='navy', lw=2, linestyle='--', label='Random Guess')
plt.xlim([0.0, 1.0])
plt.ylim([0.0, 1.05])
plt.xlabel('False Positive Rate (FPR)')
plt.ylabel('True Positive Rate (TPR)')
plt.title('Receiver Operating Characteristic (ROC) Curve')
plt.legend(loc="lower right")
plt.grid(True)
plt.show() # 在 优快云 或 Jupyter Notebook 中会显示图像

(代码运行后会生成 ROC 曲线图)

四、回归模型评估指标

回归任务的目标是预测一个连续值(如房价、股票价格、温度等)。评估回归模型的指标主要关注预测值与真实值之间的差异。

4.1 平均绝对误差 (Mean Absolute Error, MAE)

MAE 计算的是预测值与真实值之间绝对误差的平均值

M A E = 1 n ∑ i = 1 n ∣ y i − y ^ i ∣ MAE = \frac{1}{n} \sum_{i=1}^{n} |y_i - \hat{y}_i| MAE=n1i=1nyiy^i
其中, n n n 是样本数量, y i y_i yi 是第 i i i 个样本的真实值, y ^ i \hat{y}_i y^i 是模型对第 i i i 个样本的预测值。

优点

  • 直观易懂,表示平均预测误差的大小。
  • 对异常值(outliers)相对不敏感。

缺点

  • 绝对值函数在零点处不可导,可能给某些优化算法带来问题(虽然实践中影响不大)。

4.2 均方误差 (Mean Squared Error, MSE)

MSE 计算的是预测值与真实值之间误差平方的平均值

M S E = 1 n ∑ i = 1 n ( y i − y ^ i ) 2 MSE = \frac{1}{n} \sum_{i=1}^{n} (y_i - \hat{y}_i)^2 MSE=n1i=1n(yiy^i)2

优点

  • 数学性质好,便于求导和优化(是线性回归等模型的常用损失函数)。
  • 对较大的误差施加更大的惩罚(因为平方)。

缺点

  • 对异常值非常敏感:一个很大的误差会被平方放大,可能导致 MSE 过高,掩盖模型在其他样本上的表现。
  • 单位与原始数据不一致(是原始数据单位的平方)。

4.3 均方根误差 (Root Mean Squared Error, RMSE)

RMSE 是 MSE 的平方根,使得评估指标的单位与原始数据一致

R M S E = M S E = 1 n ∑ i = 1 n ( y i − y ^ i ) 2 RMSE = \sqrt{MSE} = \sqrt{\frac{1}{n} \sum_{i=1}^{n} (y_i - \hat{y}_i)^2} RMSE=MSE =n1i=1n(yiy^i)2

优点

  • 单位与原始数据相同,更易于解释。
  • 保留了 MSE 对大误差惩罚较重的特性。

缺点

  • 仍然对异常值敏感。

MAE vs. RMSE:

  • RMSE 通常大于或等于 MAE。
  • 如果 RMSE 远大于 MAE,说明存在一些具有较大误差的异常预测值。
  • 选择哪个取决于业务场景:如果大误差是不可接受的,RMSE 更合适;如果所有误差同等重要,MAE 可能更合适。

4.4 R 方 (R-squared, Coefficient of Determination)

R 方(也叫决定系数)衡量的是模型预测能够解释因变量(真实值)方差的比例。它的值域通常在 [0, 1] 之间(但也可以为负数,表示模型比直接预测平均值还差)。

R 2 = 1 − ∑ i = 1 n ( y i − y ^ i ) 2 ∑ i = 1 n ( y i − y ˉ ) 2 = 1 − M S E Var ( y ) R^2 = 1 - \frac{\sum_{i=1}^{n} (y_i - \hat{y}_i)^2}{\sum_{i=1}^{n} (y_i - \bar{y})^2} = 1 - \frac{MSE}{\text{Var}(y)} R2=1i=1n(yiyˉ)2i=1n(yiy^i)2=1Var(y)MSE
其中, y ˉ \bar{y} yˉ 是真实值的平均值, ∑ i = 1 n ( y i − y ˉ ) 2 \sum_{i=1}^{n} (y_i - \bar{y})^2 i=1n(yiyˉ)2 是真实值的总平方和(Total Sum of Squares, SST), ∑ i = 1 n ( y i − y ^ i ) 2 \sum_{i=1}^{n} (y_i - \hat{y}_i)^2 i=1n(yiy^i)2 是残差平方和(Sum of Squared Residuals, SSR)。

解读 R 方

  • R² = 1:模型完美拟合数据,所有变异都能被模型解释。
  • R² = 0:模型等同于用平均值进行预测,无法解释任何变异。
  • R² 越接近 1,表示模型的拟合优度越好。
  • R² < 0:模型表现非常差,还不如直接预测平均值。

注意:R² 容易受到特征数量的影响,即使加入无关的特征,R² 也可能略微增加。因此,有时会使用调整 R 方 (Adjusted R-squared),它考虑了模型中自变量的数量。

# Scikit-Learn 示例: 计算回归指标
from sklearn.metrics import mean_absolute_error, mean_squared_error, r2_score
from sklearn.linear_model import LinearRegression
from sklearn.datasets import make_regression
import numpy as np

# 生成模拟回归数据
X_reg, y_reg = make_regression(n_samples=1000, n_features=10, noise=20, random_state=42)
X_reg_train, X_reg_test, y_reg_train, y_reg_test = train_test_split(X_reg, y_reg, test_size=0.3, random_state=42)

# 训练一个线性回归模型
reg_model = LinearRegression()
reg_model.fit(X_reg_train, y_reg_train)
y_reg_pred = reg_model.predict(X_reg_test)

# 计算评估指标
mae = mean_absolute_error(y_reg_test, y_reg_pred)
mse = mean_squared_error(y_reg_test, y_reg_pred)
rmse = np.sqrt(mse) # 或者 mean_squared_error(..., squared=False)
r2 = r2_score(y_reg_test, y_reg_pred)

print(f"平均绝对误差 (MAE): {mae:.4f}")
print(f"均方误差 (MSE): {mse:.4f}")
print(f"均方根误差 (RMSE): {rmse:.4f}")
print(f"R 方 (R-squared): {r2:.4f}")

输出示例:

平均绝对误差 (MAE): 15.9428
均方误差 (MSE): 402.5506
均方根误差 (RMSE): 20.0637
R 方 (R-squared): 0.9594

五、模型验证:交叉验证 (Cross-Validation)

我们在前面计算评估指标时,通常是将数据分为训练集 (train set) 和测试集 (test set)。模型在训练集上学习,然后在测试集上评估性能。但这种单次划分可能存在偶然性:

  • 如果测试集恰好包含一些模型容易处理的样本,评估结果会过于乐观。
  • 如果测试集恰好包含一些模型难以处理的样本,评估结果会过于悲观。

为了得到更稳健、更可靠的模型性能评估,我们使用交叉验证

5.1 K 折交叉验证 (K-Fold Cross-Validation)

K 折交叉验证是最常用的交叉验证方法。

步骤

  1. 将原始训练数据集随机划分为 K 个大小相似的、互不相交的子集(称为“折”,Fold)。
  2. 进行 K 轮迭代:
    • 在每一轮 (i = 1 to K),选择第 i 个子集作为验证集 (Validation Set),其余 K-1 个子集作为训练集 (Training Set)
    • 在训练集上训练模型。
    • 在验证集上评估模型性能,记录评估指标(如准确率、AUC、RMSE 等)。
  3. 将 K 轮得到的评估指标进行平均(有时也看标准差),得到最终的模型性能评估结果。
K折交叉验证
交叉验证过程
划分为 5 折
原始训练数据
Fold 1
Fold 2
Fold 3
Fold 4
Fold 5
Fold 1-5
轮流作为验证集
其余为训练集
训练 5 个模型
生成 5 个评估指标
平均 5 个指标
最终评估结果

图示:K 折交叉验证流程 (K=5)

优点

  • 所有数据都参与了训练和验证,评估结果更稳定、偏差更小。
  • 有效利用了数据。

缺点

  • 计算成本是单次划分的 K 倍。

K 的选择:常用 K=5 或 K=10。较大的 K 会减少偏差但增加方差和计算时间。

5.2 分层 K 折交叉验证 (Stratified K-Fold Cross-Validation)

分类问题中,如果数据类别分布不平衡,标准的 K 折交叉验证可能导致某些折中某个类别的样本比例与整体数据差异很大,甚至缺失某个类别。

分层 K 折交叉验证 在划分数据时,会确保每个折中的类别比例与原始数据集中的类别比例大致相同。这对于不平衡数据集的评估至关重要。

对于分类任务,通常推荐使用 Stratified K-Fold。

# Scikit-Learn 示例: 使用 K 折和分层 K 折交叉验证评估模型
from sklearn.model_selection import KFold, StratifiedKFold, cross_val_score

# 使用原始分类数据 X, y

# K 折交叉验证 (K=5)
kf = KFold(n_splits=5, shuffle=True, random_state=42) # shuffle=True 建议加上,打乱数据
# cross_val_score 会自动完成 K 次训练和评估
# scoring 参数指定评估指标, 对于分类常用的有 'accuracy', 'precision', 'recall', 'f1', 'roc_auc'
# 对于回归常用的有 'neg_mean_squared_error', 'neg_mean_absolute_error', 'r2' (注意带 neg_)
scores_kfold = cross_val_score(model, X, y, cv=kf, scoring='accuracy')
print(f"K 折交叉验证 (Accuracy): {scores_kfold}")
print(f"平均准确率: {scores_kfold.mean():.4f} (+/- {scores_kfold.std() * 2:.4f})") # 通常报告均值和两倍标准差区间

# 分层 K 折交叉验证 (K=5)
skf = StratifiedKFold(n_splits=5, shuffle=True, random_state=42)
scores_skfold = cross_val_score(model, X, y, cv=skf, scoring='roc_auc') # 试试用 AUC 作为指标
print(f"\n分层 K 折交叉验证 (AUC): {scores_skfold}")
print(f"平均 AUC: {scores_skfold.mean():.4f} (+/- {scores_skfold.std() * 2:.4f})")

输出示例:

K 折交叉验证 (Accuracy): [0.84  0.87  0.845 0.825 0.86 ]
平均准确率: 0.8480 (+/- 0.0322)

分层 K 折交叉验证 (AUC): [0.9207 0.9364 0.9258 0.907  0.9359]
平均 AUC: 0.9252 (+/- 0.0216)

cross_val_score 返回一个包含 K 个分数的数组,我们可以计算其均值和标准差来评估模型的平均性能和稳定性。

六、模型选择策略

有了可靠的评估指标和验证方法,我们就可以进行模型选择了。通常有以下几种策略:

6.1 基于评估指标选择

  1. 确定核心评估指标:根据业务目标选择最重要的评估指标(或少数几个)。例如,欺诈检测可能最关心召回率或 F1 分数,而不是准确率。
  2. 交叉验证评估:对所有候选模型(不同算法或同一算法不同超参数组合)使用交叉验证(推荐 Stratified K-Fold 用于分类)计算核心评估指标的平均值和标准差。
  3. 选择最优模型:选择在核心评估指标上表现最好且性能稳定的模型。有时需要在多个指标间权衡(例如,使用 F1 综合 Precision 和 Recall)。

6.2 考虑业务需求与成本

模型选择不仅仅是看指标数字,还需要结合实际业务场景:

  • 不同错误类型的成本:预测错误的代价是什么?漏诊癌症 (FN) 的成本远高于误诊 (FP)。将正常邮件判为垃圾邮件 (FP) 的成本可能高于放过一封垃圾邮件 (FN)。需要选择能够更好控制高成本错误的模型或调整阈值。
  • 模型复杂度与可解释性:简单的模型(如逻辑回归、决策树)更容易解释,便于业务人员理解和信任。复杂的模型(如深度神经网络)可能性能更好,但像个“黑箱”。根据业务对可解释性的要求选择。
  • 预测速度与资源消耗:模型上线后,预测需要多快?消耗多少计算资源?简单的模型通常更快、更轻量。需要在性能和效率之间找到平衡。

6.3 结合 A/B 测试

最终的模型选择往往需要在线上进行 A/B 测试。将新模型与旧模型(或基线模型)同时部署,将用户流量随机分配给它们,直接比较它们在真实业务指标(如点击率、转化率、用户留存率)上的表现。这是检验模型实际效果的黄金标准。

七、实战:使用 Scikit-Learn 进行评估与选择

假设我们正在处理一个二分类问题,数据可能存在不平衡。我们尝试了逻辑回归 (LR) 和随机森林 (RF) 两种模型。

import numpy as np
from sklearn.datasets import make_classification
from sklearn.model_selection import train_test_split, StratifiedKFold, cross_validate
from sklearn.linear_model import LogisticRegression
from sklearn.ensemble import RandomForestClassifier
from sklearn.metrics import make_scorer, accuracy_score, precision_score, recall_score, f1_score, roc_auc_score

# 1. 准备数据 (假设存在轻微不平衡)
X, y = make_classification(n_samples=2000, n_features=20, n_informative=15, n_redundant=5,
                           weights=[0.8, 0.2], # 80% 负类, 20% 正类
                           flip_y=0.05, random_state=42, class_sep=0.8)

# 划分训练集和保留的最终测试集 (Hold-out set)
# 交叉验证在训练集内部进行
X_train_val, X_test, y_train_val, y_test = train_test_split(X, y, test_size=0.2, random_state=42, stratify=y)

# 2. 定义候选模型
lr_model = LogisticRegression(solver='liblinear', random_state=42)
rf_model = RandomForestClassifier(n_estimators=100, random_state=42)

# 3. 定义评估指标 (选择多个关注的指标)
scoring = {
    'accuracy': make_scorer(accuracy_score),
    'precision': make_scorer(precision_score, zero_division=0), # 处理 precision 为 0 的情况
    'recall': make_scorer(recall_score),
    'f1': make_scorer(f1_score),
    'roc_auc': make_scorer(roc_auc_score, needs_proba=True) # AUC 需要概率
}

# 4. 使用分层 K 折交叉验证进行评估 (在 X_train_val, y_train_val 上)
skf = StratifiedKFold(n_splits=5, shuffle=True, random_state=42)

print("评估 Logistic Regression:")
# cross_validate 可以同时返回多个指标的 K 折结果
lr_scores = cross_validate(lr_model, X_train_val, y_train_val, cv=skf, scoring=scoring)
for metric, scores in lr_scores.items():
    if metric.startswith('test_'): # 我们关心验证集上的分数 ('test_' 前缀是 cross_validate 的约定)
        print(f"  {metric[5:]}: {np.mean(scores):.4f} (+/- {np.std(scores) * 2:.4f})")


print("\n评估 Random Forest:")
rf_scores = cross_validate(rf_model, X_train_val, y_train_val, cv=skf, scoring=scoring)
for metric, scores in rf_scores.items():
     if metric.startswith('test_'):
        print(f"  {metric[5:]}: {np.mean(scores):.4f} (+/- {np.std(scores) * 2:.4f})")

# 5. 分析结果,选择模型
# - 随机森林在 Accuracy, Recall, F1, AUC 上似乎都优于逻辑回归。
# - 逻辑回归的 Precision 稍高,但 Recall 较低,导致 F1 不如随机森林。
# - 考虑到 AUC 和 F1 的综合表现,随机森林可能是更好的选择。
# (实际选择还需结合业务需求和对 Precision/Recall 的侧重)

# 6. (可选) 在完整的训练验证集上训练最终模型,并在保留的测试集上进行最终评估
final_model = RandomForestClassifier(n_estimators=100, random_state=42) # 假设选择 RF
final_model.fit(X_train_val, y_train_val)
y_final_pred = final_model.predict(X_test)
y_final_pred_proba = final_model.predict_proba(X_test)[:, 1]

print("\n在最终测试集上的性能 (Random Forest):")
print(f"  Accuracy: {accuracy_score(y_test, y_final_pred):.4f}")
print(f"  Precision: {precision_score(y_test, y_final_pred, zero_division=0):.4f}")
print(f"  Recall: {recall_score(y_test, y_final_pred):.4f}")
print(f"  F1 Score: {f1_score(y_test, y_final_pred):.4f}")
print(f"  AUC: {roc_auc_score(y_test, y_final_pred_proba):.4f}")

(代码运行后会输出 LR 和 RF 在交叉验证中的各项平均指标及标准差,以及最终选定模型在测试集上的表现)

八、总结

科学的模型评估与选择是机器学习流程中至关重要的一环。脱离评估谈模型优劣是毫无意义的。本文我们系统学习了:

  1. 评估的必要性:量化性能、比较模型、诊断问题、评估泛化能力。
  2. 分类模型核心指标
    • 混淆矩阵:评估的基础。
    • 准确率 (Accuracy):直观但对不平衡数据有误导性(陷阱!)。
    • 精确率 (Precision):关注预测为正的准确性,避免误报 (FP)。
    • 召回率 (Recall):关注实际为正的覆盖率,避免漏报 (FN)。
    • F1 分数 (F1 Score):精确率和召回率的调和平均,兼顾两者。
    • ROC 曲线与 AUC 值:衡量模型在不同阈值下的区分能力,对不平衡数据更鲁棒。
  3. 回归模型核心指标
    • MAE:平均绝对误差,对异常值不敏感。
    • MSE/RMSE:均方(根)误差,对大误差惩罚更重,对异常值敏感。
    • R 方 (R-squared):解释方差的比例,衡量拟合优度。
  4. 模型验证方法
    • K 折交叉验证 (K-Fold):提供更稳健的性能评估。
    • 分层 K 折交叉验证 (Stratified K-Fold):分类任务(尤其是不平衡数据)的首选。
  5. 模型选择策略
    • 基于核心评估指标的交叉验证结果。
    • 结合业务需求、错误成本、可解释性、效率等。
    • 通过 A/B 测试 进行最终验证。
  6. Scikit-Learn 实战:演示了如何使用 sklearn.metrics 计算各种指标,以及如何通过 cross_validateStratifiedKFold 进行可靠的模型评估和比较。

希望通过今天的学习,大家能够掌握这些关键的评估与选择技术,为后续的模型调优(Day 18: 超参数调优)和项目实践打下坚实的基础!不再被单一的准确率所迷惑,能够根据具体任务选择最合适的评估标准和模型。


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

吴师兄大模型

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值