You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

如何将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

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.07.30 22:09:24