小编spo*_*shr的帖子

GridSearchCV 结果热图

我正在尝试为 sklearn 的 GridSearchCV 结果生成热图。我喜欢sklearn-evaluation的一点是它非常容易生成热图。然而,我遇到了一个问题。当我将参数指定为 None 时,例如

max_depth = [3, 4, 5, 6, None]
Run Code Online (Sandbox Code Playgroud)

生成热图时,它显示错误:

TypeError: '<' not supported between instances of 'NoneType' and 'int'
Run Code Online (Sandbox Code Playgroud)

有什么解决方法吗?我找到了其他生成热图的方法,例如使用 matplotlib 和 seaborn,但没有什么能像 sklearn-evalutaion 那样提供漂亮的热图。

在此输入图像描述

python matplotlib scikit-learn seaborn sklearn-pandas

4
推荐指数
1
解决办法
1701
查看次数