有没有办法将H2O部分相关的绘图结果保存为数据帧？

Question

我正在尝试使用以下代码创建部分相关绘图

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

有没有办法可以把mean_resp之类的输出保存到数据帧中？在

Answer 1

如果将plot参数设置为True，则partial_plot()方法返回元素为h2o.two_dim_table.H2OTwoDimTable类型的列表或列表和绘图（请参阅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列，可以这样做

^{pr2}$

或者

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