时间戳

首页 > 解决方案 > 从 sklearn.model_selection.GridSearchCV 获取 keyerror

python - 从 sklearn.model_selection.GridSearchCV 获取 keyerror

问题描述

我正在尝试使用GridSearchCV实现决策树分类器。实施后我试图访问cv_results_.mean_train_score但我得到关键错误。

    tuned_parameters = [{'max_depth': [1, 5, 10, 25, 50, 75, 100, 150, 250, 500, 750, 1000], 
                         'min_samples_split' : [5, 10, 25, 50, 75, 150, 250, 500]}] 
    cv_timeSeries = TimeSeriesSplit(n_splits=4).split(X_train)
    base_estimator = DecisionTreeClassifier(class_weight='balanced') 
    gsearch_cv = GridSearchCV(estimator=base_estimator, 
                              param_grid=tuned_parameters, 
                              cv=cv_timeSeries, 
                              scoring='roc_auc', 
                              n_jobs=-1)
    gsearch_cv.fit(X_train, y_train)

当我试图访问 gsearch_cv 的所有键时，我无法找到字典键 mean_train_score。

标签： pythonmachine-learningdecision-treegridsearchcv

解决方案

在 GridSearchCV 中添加以下参数

GridSearchCV(return_train_score=True)

推荐阅读