GridSearchCV - 每次迭代保存结果

9
我正在使用GridSearchCV,在每次迭代后,我想将clf.cv_results_属性保存到文件中(以防进程在中途崩溃)。
我试图寻找解决方案,但是我无法弄清楚。
任何帮助都将不胜感激。

请添加一些示例代码和数据集的元素,以便帮助。 - Espoir Murhabazi
1个回答

7

其中一种方法是将网格搜索的verbose参数设置为大于0的整数,如10或更高,它会将每次迭代的结果打印到控制台。

有了这个参数,您的输出应该被打印或记录到控制台中,然后按照答案的说明,将记录下来的结果放入文件中。

基本上,在运行GridSearch之前添加以下内容:

import sys
old_stdout = sys.stdout

log_file = open("message.log","w")

sys.stdout = log_file

在运行GridSearch之后,您应该使用以下代码关闭所有资源:

sys.stdout = old_stdout
log_file.close()

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接