add return_train_score argument

KangchengHou · Nov 8, 2019 · 4cd1e14 · 4cd1e14
1 parent 59c3120
commit 4cd1e14
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/search.py b/search.py
@@ -22,7 +22,7 @@ def search(dataset, data_dir):
     svc = SVC(kernel='precomputed', cache_size=16000, max_iter=5e5)
     clf = GridSearchCV(svc, {'C' : C_list}, 
                 cv=zip(train_fold_idx, test_fold_idx),
-                n_jobs=80, verbose=0)
+                n_jobs=80, verbose=0, return_train_score=True)
     clf.fit(gram, labels)
     df = pd.DataFrame({'C': C_list, 
                        'train': clf.cv_results_['mean_train_score'],