模型的评估与选择——交叉验证(基于Python实现)
模型的评估与选择——交叉验证
1. 交叉验证-模拟1
交叉验证是一种用于估计机器学习模型性能的统计方法。它涉及将数据划分为子集,在一些子集上训练模型,并在剩余的子集上验证模型。这个过程会重复多次,以确保模型的性能是一致的,并且不依赖于特定的数据子集。
以下是最常见的交叉验证技术的简要概述:
-
K折交叉验证:将数据分成
k
个大小相等的折叠。模型在k-1
个折叠上训练,并在剩余的一个折叠上验证。这个过程重复k
次,每个折叠恰好用一次作为验证数据。 -
留一法交叉验证(LOOCV):k折交叉验证的一种特殊情况,其中
k
等于数据点的数量。每个数据点恰好用一次作为验证集,模型在剩余的数据上训练。 -
分层K折交叉验证:类似于k折交叉验证,但折叠的创建方式使得每个折叠中的类别比例与原始数据集中的比例相同。这对于不平衡数据集特别有用。
-
时间序列交叉验证:用于时间序列数据,其中数据点的顺序很重要。数据被分成训练和验证集,方式是尊重时间顺序。
以下是使用scikit-learn
在Python中进行k折交叉验证的示例:
import numpy as np
from sklearn.model_selection import KFold
from sklearn.linear_model import LinearRegression
from sklearn.metrics import mean_squared_error# 样本数据
X = np.random.rand(100, 5)
y = np.random.rand(100)# K折交叉验证
kf = KFold(n_splits=5)
model = LinearRegression()
mse_scores = []for train_index, test_index in kf.split(X):X_train, X_test = X[train_index], X[test_index]y_train, y_test = y[train_index], y[test_index]model.fit(X_train, y_train)y_pred = model.predict(X_test)mse_scores.append(mean_squared_error(y_test, y_pred))print("每个折叠的均方误差:", mse_scores)
print("平均均方误差:", np.mean(mse_scores))
这段代码演示了如何执行k折交叉验证来评估线性回归模型的性能。
#生成数据
# 模拟1
import random
import numpy as np
import matplotlib.pyplot as plt
from sklearn.neighbors import KNeighborsClassifiernq=5000
Q=np.empty((50,nq))
for i in range(50):for j in range(nq):Q[i,j]=random.gauss(0,1)
N1=np.zeros(25)
N2=np.ones(25)
N=np.hstack((N1,N2))
random.shuffle(N) #打乱成数组或列表
# 交叉验证
b=np.empty(nq)
nc=100
for i in range(nq):b[i]=np.corrcoef(N,Q[:,i]