转换H2O模型结果的熊猫数据帧

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) 

现在我想将这些结果保存到数据帧大熊猫。

我试图做到这一点，像这样：

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'> 

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

type(m.varimp()) 
# <type 'list'> 

所以，看看在m输出，决定你想要什么，在dir(m)与方法的名字称呼它，然后根据需要转换成数据帧。