试图寻找分类器的最佳参数。
以下是我的代码:
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)))
但我想绘制分类器参数的平均准确率图表。
请帮帮我,提前谢谢!
回答:
由于GridSearchCV
进行的是详尽搜索,你可能会得到一个结果矩阵,即每个参数组合与准确率相关联。你可以从中选择你想要绘制的内容
希望这样的矩阵存在,并且可以通过GridSearchCV
的属性cv_results_
获取
即你可以这样做:
result_matrix = cv.cv_results_
祝你绘图愉快pltrdy