轉換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)與方法的名字稱呼它，然後根據需要轉換成數據幀。