KeyError Нет при запуске .fit на gridsearchcv - PullRequest
Новая
       15

KeyError Нет при запуске .fit на gridsearchcv

0 голосов
/ 12 октября 2019

У меня есть параметры дерева настроек и стратификация kfold. Теперь используйте GridSearchCV с DecisioonTreeClassifier и запустите метод .fit. Однако я получаю сообщение об ошибке KeyError: None

с помощью записной книжки Jupyter и ранее запускал GridSearchCV без ошибок, не уверенный, почему это происходит с ошибкой для этого кадра данных 

tree = DecisionTreeClassifier(np.random.seed(17), class_weight = 'balanced')

max_depth_values = [5, 6, 7, 8, 9]
max_features_values = [4, 5, 6, 7]
tree_params = {'max_depth': max_depth_values, 'max_features': max_features_values}

skf = StratifiedKFold(n_splits=5, shuffle=True, random_state = np.random.seed(17))

tree_grid = GridSearchCV(tree, tree_params, cv=skf, scoring='roc_auc')

tree_grid.fit(X, y)

The code fails with the below error and I am not sure whats causing this

---------------------------------------------------------------------------
KeyError                                  Traceback (most recent call last)
<ipython-input-45-3ea31d6abb73> in <module>
      3 tree_grid = GridSearchCV(tree, tree_params, cv=skf, scoring='roc_auc')
      4 
----> 5 tree_grid.fit(X, y)

KeyError: None
...