LearningCurve检视过拟合

from sklearn.model_selection import learning_curve
from sklearn.datasets import load_digits
from sklearn.svm import SVC  # Support Vector Classifier
import matplotlib.pyplot as plt
import numpy as np

"""加载digits数据集，其包含的是手写体的数字，从0到9。数据集总共有1797个样本，每个样本由64个特征组成， 
分别为其手写体对应的8×8像素表示，每个特征取值0~16。"""
digits = load_digits()
x = digits.data
y = digits.target

train_sizes, train_loss, test_loss = learning_curve(
    SVC(gamma=0.001), x, y, cv=10, scoring="neg_mean_squared_error",
    train_sizes=[0.1, 0.25, 0.5, 0.75, 1]
)
# 平均每一轮所得到的平均方差(共5轮，分别为样本10%、25%、50%、75%、100%)
train_loss_mean = -np.mean(train_loss, axis=1)
test_loss_mean = -np.mean(test_loss, axis=1)
# 可视化图形:
plt.plot(train_sizes,train_loss_mean,'o-', color="r",label="Training")
plt.plot(train_sizes,test_loss_mean,'o-',color='g',label="Cross-validation")
plt.xlabel("Training examples")
plt.ylabel("Loss")
plt.legend(loc="best")
plt.show()