H2O目标均值归因可以为每个分类变量级别输出映射表吗?

时间:2019-03-23 03:18:10

标签: python h2o

我正在从h2o.targetencoder学习H2O TargetEncoder函数。我想知道,是否有一种方法可以导出分类变量的每个级别的映射表?

例如,如果我们有2个针对50个美国州和5个工作日的分类变量。 TargetEncoder可以导出带有50个州和5个工作日的映射数值的结果表吗?

1 个答案:

答案 0 :(得分:0)

以示例here为例,您可以通过以下方式调用targetEncoder.fit(ext_train)后访问映射:

encodingMapFramesKeys = list(map(lambda x: h2o.get_frame(x['key']['name']), targetEncoder._encodingMap.frames))

encodingMapFramesKeys[0].describe()
encodingMapFramesKeys[1].describe()