将h2o模型结果转换为Pandas数据帧

Question

我在水里训练一个模型，像这样：

import h2o

h2o.init()
trainFrame = h2o.import_file(path = "C:/train.csv")

train, test = trainFrame.split_frame([0.8])

x = ["A", "B", "C"]
y = "Target"

m = h2o.estimators.H2ORandomForestEstimator(model_id="RF_defaults")
m.train(x, y, train)

现在我想把这些结果保存到pandas数据框中。在

我正试着这样做：

m.as_data_frame(pandas = True)

但这又回来了：

AttributeError: type object 'H2ORandomForestEstimator' has no attribute 'as_data_frame'

Answer 1

模型对象m中包含大量信息。使用dir(m)查看所有可用组件。在

在某些情况下，m的一个组件已经作为数据帧存储，例如评分历史：

type(m.scoring_history())
# <class 'pandas.core.frame.DataFrame'>

在其他情况下，组件将是一个值列表，例如变量重要性：

^{2}$

所以，看看m的输出，决定您想要什么，用dir(m)中的方法名调用它，然后根据需要转换为数据帧。在