如何绘制分类器参数误差图

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

以下是我的代码:

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_获取

来自GridSearchCV的文档:enter image description here

即你可以这样做:

result_matrix = cv.cv_results_

祝你绘图愉快pltrdy

Related Posts

L1-L2正则化的不同系数

我想对网络的权重同时应用L1和L2正则化。然而,我找不…

使用scikit-learn的无监督方法将列表分类成不同组别,有没有办法?

我有一系列实例,每个实例都有一份列表,代表它所遵循的不…

f1_score metric in lightgbm

我想使用自定义指标f1_score来训练一个lgb模型…

通过相关系数矩阵进行特征选择

我在测试不同的算法时,如逻辑回归、高斯朴素贝叶斯、随机…

可以将机器学习库用于流式输入和输出吗?

已关闭。此问题需要更加聚焦。目前不接受回答。 想要改进…

在TensorFlow中,queue.dequeue_up_to()方法的用途是什么?

我对这个方法感到非常困惑,特别是当我发现这个令人费解的…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注