GridSearchCV не извлекает mean_test_score в colab

Question

Я пробежал randomForestClassifier, используя GridSearchCV в колабе.Однако myGridSearchCV не возвращает mean_test_score и показывает, что его не существует.Когда я пытался проверить clf.cv результаты , это не значит mean_test_score. Это работает в ноутбуке jupyter, а не в colab, и я недавно столкнулся с этой проблемой.

[Parallel(n_jobs=1)]: Done 120 out of 120 | elapsed: 25.3min finished
---------------------------------------------------------------------------
KeyError                                  Traceback (most recent call last)
<ipython-input-128-1d8f6c95affb> in <module>()
      7 clf.fit(X_tr, y_train)
      8 
----> 9 train_auc= clf.cv_results_['mean_train_score']
     10 cv_auc = clf.cv_results_['mean_test_score']

KeyError: 'mean_train_score'

Answer 1

mean_train_score отсутствует, вы можете проверить это с помощью

scores = rf.cv_results_

pd.DataFrame(scores).head()

рф - это ваша сетка поиска

вы получите сообщение об ошибке, подобное приведенному ниже

KeyError: 'mean_train_score'

решение

set return_train_score = True в gridsearchcv

rf = GridSearchCV(rf, parameters, cv=n_folds, scoring="accuracy",return_train_score=True)

Answer 2

На самом деле вы получаете mean_train_score ошибку ключа.Просто установите return_train_score=True в gridsearchcv параметры, и он сделает эту работу.