获取一键编码的H2OFrame

时间:2019-10-10 15:25:32

标签: python machine-learning h2o shap

我正在H2O中使用xgboost进行二进制分类任务。数据集具有几个分类特征,模型在训练期间对其应用了一次热编码。

现在,我想使用SHAP(https://stackoverflow.com/a/58372842/2914140)在本地解释预测。为此,最好将数据帧与一键编码的列和值一起使用。但是,我似乎找不到从H2O模型中获取此信息的方法。

我可能可以手动重新创建一键编码,但是也许有人知道更快的解决方案?

1 个答案:

答案 0 :(得分:1)

我们已经有ticket open了一段时间了,但是由于对此功能的需求增加,我们将很快重新访问它。目前,您必须使用as_data_frame()方法将H2OFrame转换为Pandas DataFrame,然后应用以下solutions之一。