Result of GridSearchCV as table

Question

I did grid search + crossvalidation on a SVM with RBF kernel to find optimal value of parameters C and gamma using the class GridShearchCV. Now I would like to get the result in a tabular format like

C/gamma 1e-3 1e-2 1e3
0.1      0.2  ..  0.3
1        0.9
10       ..   
100      ..

where cells contain accuracy score for that couple of parameters values.

Or at least, if first solution is not possible, something easier like

C    gamma  accuracy
0.1  1e-4      0.2 
...

I am not very skilled in Python, so I don't know where to start. Could you give me some method to do this kind of representations? The best solution would be to have the table as a plot but also a simple print in console in those formats would be fine. Thank you in advance.

`import pandas as pd;df = pd.Dataframe(grid.cv_results_)` ? Couldn't test, did it work? — Shihab Shahriar Khan, Nov 13 '19 at 15:10
Hi @Gianluca Amprimo, if the below solution worked for you, I kindly request you to accept it as a verified answer as it would help others who might visit this question later. — Parthasarathy Subburaj, Nov 26 '19 at 14:58

score 10 · Answer 1 · edited Nov 23 '22 at 13:03

You could make use of the cv_results_ attribute of the gridsearchCV object as shown below:

from sklearn import svm, datasets
from sklearn.model_selection import GridSearchCV
iris = datasets.load_iris()
parameters = {'kernel':('linear', 'rbf'), 'C':[1, 10]}
svc = svm.SVC(gamma="scale")
clf = GridSearchCV(svc, parameters, cv=5)
clf.fit(iris.data, iris.target)

Now you use clf.cv_results_

{'mean_fit_time': array([0.00049248, 0.00051575, 0.00051174, 0.00044131]),
 'mean_score_time': array([0.0002739 , 0.00027657, 0.00023718, 0.00023627]),
 'mean_test_score': array([0.98      , 0.96666667, 0.97333333, 0.98      ]),
 'param_C': masked_array(data=[1, 1, 10, 10],
              mask=[False, False, False, False],
        fill_value='?',
             dtype=object),
 'param_kernel': masked_array(data=['linear', 'rbf', 'linear', 'rbf'],
              mask=[False, False, False, False],
        fill_value='?',
             dtype=object),
 'params': [{'C': 1, 'kernel': 'linear'},
  {'C': 1, 'kernel': 'rbf'},
  {'C': 10, 'kernel': 'linear'},
  {'C': 10, 'kernel': 'rbf'}],
 'rank_test_score': array([1, 4, 3, 1], dtype=int32),
 'split0_test_score': array([0.96666667, 0.96666667, 1.        , 0.96666667]),
 'split1_test_score': array([1.        , 0.96666667, 1.        , 1.        ]),
 'split2_test_score': array([0.96666667, 0.96666667, 0.9       , 0.96666667]),
 'split3_test_score': array([0.96666667, 0.93333333, 0.96666667, 0.96666667]),
 'split4_test_score': array([1., 1., 1., 1.]),
 'std_fit_time': array([1.84329827e-04, 1.34653950e-05, 1.26220210e-04, 1.76294378e-05]),
 'std_score_time': array([6.23956317e-05, 1.34498512e-05, 3.57596078e-06, 4.68175419e-06]),
 'std_test_score': array([0.01632993, 0.02108185, 0.03887301, 0.01632993])}

You can make use of the params and the mean_test_score for constructing the dataframe you are looking using the below command:

pd.concat([pd.DataFrame(clf.cv_results_["params"]),pd.DataFrame(clf.cv_results_["mean_test_score"], columns=["Accuracy"])],axis=1)

And your final dataframe looks like

    C   kernel  Accuracy
0   1   linear  0.980000
1   1   rbf     0.966667
2   10  linear  0.973333
3   10  rbf     0.980000

keramat · Answer 2 · 2021-02-21T12:57:11.867

8

Perhaps easier:

pd.DataFrame({'param': clf.cv_results_["params"], 'acc': clf.cv_results_["mean_test_score"]})

or:

df = pd.DataFrame(clf.cv_results_)

edited Feb 21 '21 at 12:57

answered Feb 21 '21 at 12:32

keramat

4,328
6
25
38

@GianlucaAmprimo this should be the selected answer. Albeit both functionally work, this answer requires less code. – Manuel Fuchs Jun 21 '22 at 06:40

Result of GridSearchCV as table

2 Answers2