我正在尝试为 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 那样提供漂亮的热图。