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

Question

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

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

是否可以将诸如mean_resp之类的输出保存到数据帧中？

Answer 1

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