如何在分类器的参数上绘制错误图

Question

试图找到分类器的最佳参数。

关注我的代码：

kf = StratifiedKFold(n_splits = 5, random_state=42)
param = {"criterion":["gini", "entropy"], "max_features":[1, 2, 3, 4, 5, 10,      15, 20, 25, 30, 35, 40, "log2", "auto"],
         "max_depth":[2, 4, 8, 16, 32, 64], "class_weight":[None, "balanced"]}
cv = GridSearchCV(RandomForestClassifier(), param, scoring="accuracy", n_jobs = -1, cv=kf)
cv.fit(X_train, y_train);
print("Best score (accuracy): ", cv.best_score_)
print("Best params: ", cv.best_params_)
print("Score on X_test: ", accuracy_score(y_test, cv.predict(X_test)))

但我想在分类器的参数上绘制分类器平均值的图表。

请帮帮我， 提前谢谢！

Answer 1

当GridSearchCV正在进行穷举搜索时，您可能会有一个结果矩阵，即每个参数组合与准确性相关联。然后，您可以从此

中选择要绘制的内容

希望存在这样的矩阵，并且可以使用GridSearchCV的属性cv_results_来检索

来自GridSearchCV's doc：

即。你可以这样做：

result_matrix = cv.cv_results_

快乐的策划 pltrdy