python - 有没有一种方法可以将H2O部分依赖图的结果保存为数据框？

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

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

有没有办法将诸如mean_resp之类的输出保存到数据帧中？

最佳答案

partial_plot()方法返回一个列表，其中元素的类型为h2o.two_dim_table.H2OTwoDimTable；如果设置了plot参数= True，则该列表和图表（请参阅api docs了解有关参数和返回类型的更多信息）。

看到这样做：

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

选择与感兴趣的pdp列对应的H2OTwoDimTable之后，您可以选择“ mean_response”列，也可以将H2OTwoDimTable转换为熊猫数据帧，然后从中选择mean_resp。

因此，例如，要获取“ var1”的mean_response列，您可以

rf_pdp[0]["mean_response"]

要么

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