GridSearchCV - 每次迭代保存结果

时间:2018-01-02 08:54:21

标签: python machine-learning scikit-learn grid-search hyperparameters

我正在使用GridSearchCV,并且在每次迭代后我想将clf.cv_results_属性保存到文件中(以防过程在中间崩溃)。

我尝试寻找解决方案,但我无法理解。

任何帮助将不胜感激。

1 个答案:

答案 0 :(得分:3)

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

这样,您的输出应打印或记录到您的控制台, 然后按照thisthis问题的答案查看如何将记录的结果存入文件

基本上,在运行GirdSearch之前把它放在:

import sys
old_stdout = sys.stdout

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

sys.stdout = log_file

然后在运行GirdSearch之后,你应该用这个关闭所有资源:

sys.stdout = original
f.close()