python - 使用 cross_val_score 计算训练分数

Question

我正在使用 cross_val_score 来计算回归量的平均分数。这是一个小片段。

from sklearn.linear_model import LinearRegression
from sklearn.model_selection import cross_val_score 

cross_val_score(LinearRegression(), X, y_reg, cv = 5)

使用它我得到了一系列分数。我想知道验证集上的分数(在上面的数组中返回)与训练集上的分数有何不同，以了解我的模型是过拟合还是欠拟合。

有没有办法用 cross_val_score 对象做到这一点？

Answer 1

您可以使用 cross_validate 而不是 cross_val_score
根据 doc :

The cross_validate function differs from cross_val_score in two ways -

• It allows specifying multiple metrics for evaluation.
• It returns a dict containing training scores, fit-times and score-times in addition to the test score.