Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Computing training score using cross_val_score

I am using cross_val_score to compute the mean score for a regressor. Here's a small snippet.

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

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

Using this I get an array of scores. I would like to know how the scores on the validation set (as returned in the array above) differ from those on the training set, to understand whether my model is over-fitting or under-fitting.

Is there a way of doing this with the cross_val_score object?

like image 402
Iwan Thomas Avatar asked Jun 21 '17 15:06

Iwan Thomas


1 Answers

You can use cross_validate instead of cross_val_score
according to 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.
like image 150
FooBee Avatar answered Sep 28 '22 15:09

FooBee