如何将H2O GridSearch网格搜索结果保存为CSV文件?
解决H2O网格搜索结果保存为CSV的问题
你遇到的错误是因为h2o.export_file仅支持导出H2OFrame、Pandas DataFrame等结构化数据对象,而grid_perf是H2OGridSearch类型的对象,无法直接传入该函数。以下是两种可行的保存方法:
方法一:转为Pandas DataFrame后保存
利用as_data_frame()方法将网格搜索结果转换为Pandas DataFrame,再用Pandas自带的to_csv保存:
# 将网格搜索结果转换为Pandas DataFrame grid_results_df = grid_perf.as_data_frame() # 保存为CSV文件,index=False避免写入行索引 grid_results_df.to_csv('grid_search_results.csv', index=False)
方法二:转为H2OFrame后用h2o.export_file保存
先把结果转为H2OFrame,再调用H2O的导出函数:
# 将Pandas DataFrame转为H2OFrame grid_results_h2o = h2o.H2OFrame(grid_perf.as_data_frame()) # 导出为CSV,force=True覆盖已存在的文件 h2o.export_file(grid_results_h2o, 'grid_search_results.csv', force=True)
两种方法都会完整保存所有超参数组合以及对应的模型评估指标(如你指定排序的AUC),无需手动整理数据结构。
内容的提问来源于stack exchange,提问作者Adam
相关产品推荐
相关产品推荐

