有没有办法将 H2O 部分依赖图结果保存为数据框?

Is there a way to save H2O partial dependent plot results as a data frame?

我正在尝试使用以下代码创建部分依赖图

rf_pdp = rf_model .partial_plot(data = htest, cols = ['var1', 'var2', 'var3'], plot=True)
rf_pdp 

有什么方法可以将 mean_resp 之类的输出保存到数据框中吗?

partial_plot() 方法 return 是一个列表,其中元素的类型是 h2o.two_dim_table.H2OTwoDimTable 或者一个列表和一个绘图,如果您设置 plot 参数 = True(请参阅api docs 以了解有关参数和 return 类型的更多信息。

看这个:

type(rf_pdp) # should return a list
type(rf_pdp[0]) # should return h2o.two_dim_table.H2OTwoDimTable

一旦您 select 编辑了与感兴趣的 pdp 列对应的 H2OTwoDimTable,您可以 select "mean_response" 列,或者您可以将 H2OTwoDimTable 转换为 pandas数据框和 select 来自那里的 mean_resp。

例如,要获取 "var1 " 的 mean_response 列,您可以执行

rf_pdp[0]["mean_response"]

rf_pdp[0].as_data_frame()['mean_response']