GridSearchCV - 每次迭代保存结果

Question

我正在使用GridSearchCV，每次迭代后我想将clf.cv_results_属性保存到一个文件中（以防万一该过程在中间崩溃）。

我试图寻找解决方案，但我就是想不通。

任何帮助将不胜感激。

Answer 1

一种方法是将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()