交叉验证 cross validation 与 K-fold Cross Validation K折叠验证

交叉验证是机器学习中验证模型稳定性的重要手段,它通过将数据集分成训练集和验证集来评估模型。K折交叉验证是一种常见方法,将数据分为K个部分,每次用K-1部分训练模型,剩余部分验证,循环进行。在实际操作中,尤其在类别不平衡的问题中,需确保每个折叠中类别的分布接近原始数据,以增强模型的泛化能力和鲁棒性。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

在这里插入图片描述
交叉验证,cross validation是机器学习中非常常见的验证模型鲁棒性的方法。其最主要原理是将数据集的一部分分离出来作为验证集,剩余的用于模型的训练,称为训练集。模型通过训练集来最优化其内部参数权重,再在验证集上检验其表现。

比较常见的交叉验证方法K折叠交叉验证,(K-fold Cross Validation)如下图所示,我们将数据分为K个部分,其中K-1个部分作为训练集,剩余的作为验证集。每个epoch循环,都选取不一样的一个部分作为验证集。

在这里插入图片描述
在使用K折叠或者别的交叉验证的方法时,我们需要注意的是,在很多问题中,类与类在数据集中的分布不一定是均匀的。因此,我们往往需要在折叠的过程中使得每个fold都有着相同或至少近似的类的分布。

例如在下列性别二分类问题中,原始数据中女性与男性的样本数量如下。那么我们在划分K折叠的时候,也要将这个分布情况在每个折叠上还原,即保证在每个折叠上女性与男性的样本数量的比例与整体数据的比例是一致的。如果没有满足这个条件,在上述例子中,很可能出现训练时模型过多得受到了了男性样本得影响,而导致其在女性样本为主的验证集上表现很差。因此,保证K折叠每一个部分的类分布是均匀的,也能在一定程度上提升训练后模型的鲁棒性。
在这里插入图片描述

在这里插入图片描述

Python中,使用sklearn库进行3折交叉验证(3-Fold Cross Validation)是一个常见的机器学习模型评估过程。下面是一个简单的示例,展示如何使用`KFold`函数实现这个过程: ```python # 首先,假设我们有一个数据集 `df`,特征列名为 'X',目标变量列名为 'y' from sklearn.model_selection import train_test_split, KFold from sklearn.linear_model import LogisticRegression # 数据预处理,分割训练集测试集(这里假设80%的数据用于训练,20%用于测试) X_train, X_test, y_train, y_test = train_test_split(df['X'], df['y'], test_size=0.2, random_state=42) # 创建KFold实例,这里是3折交叉验证 kfold = KFold(n_splits=3, shuffle=True, random_state=42) # 对每个训练集进行独立的学习验证 for train_index, val_index in kfold.split(X_train): # 训练集验证集 X_train_fold, X_val_fold = X_train[train_index], X_train[val_index] y_train_fold, y_val_fold = y_train[train_index], y_train[val_index] # 创建并拟合模型(这里以逻辑回归为例) model = LogisticRegression() model.fit(X_train_fold, y_train_fold) # 在验证集上进行评估,例如计算准确率 predictions = model.predict(X_val_fold) accuracy = accuracy_score(y_val_fold, predictions) print(f"Validation fold accuracy: {accuracy}") # 最终,你可以选择所有折叠的平均精度作为模型性能指标 average_accuracy = sum(accuracy) / len(accuracy) print(f"Average 3-fold CV accuracy: {average_accuracy}") ```
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值