После обучения модели с помощью инструмента autoML H2O я могу увидеть важность переменной с помощью saved_model.varimp_plot(). Мне любопытно, какова техническая часть функции, которую, как утверждает H2O, делает.

Пробую простые строчки примеров кода в документации H2O.

import h2o
h2o.init()

train_data = h2o.import_file("../full_data.csv")
test_data = h2o.import_file("../201810_pca.csv")

from h2o.automl import H2OAutoML
y = "Label"
x = ['feature0','feature1','feature2','feature3','feature4','feature5','feature6','feature7','feature8','feature9','feature10',
'feature11','feature12','feature13','feature14','feature15','feature16','feature17','feature18','feature19','feature20',
'feature21','feature22','feature23','Amount','DateTime']


aml = H2OAutoML(max_models = 100, max_runtime_secs=100000, seed = 1)
aml.train(x = x, y = y, training_frame = train_data)

lb = aml.leaderboard
lb.head()
lb.head(rows=lb.nrows) # Entire leaderboard

preds = aml.predict(test_data)
h2o.save_model(aml.leader, path = "./Saved_Models")


saved_model = h2o.load_model("./Saved_Models/XGBoost_2_AutoML_20191018_174201")

training_frame = your_model.actual_params['training_frame'] #The part gives error
print(training_frame)

Как узнать, какие функции используются в обученной модели? Я хотел бы посмотреть, извлекает ли H2O и добавляет новые функции или нет.

Я использовал my_training_frame = your_model.actual_params['training_frame'], как указано в другой вопрос, но он дает ошибку:« TypeError: объект 'property' не имеет атрибута 'getitem' ».

0
Ege 23 Окт 2019 в 16:35

1 ответ

Лучший ответ

Quick Note H2O.ai предлагает несколько продуктов. Платформа с открытым исходным кодом называется H2O-3 и содержит алгоритм AutoML. AutoML в настоящее время не занимается разработкой функций для вас. Если вам нужна автоматическая разработка функций, вы можете подумать о продукте H2O Driverless-AI .

Что касается ошибки, которую вы видите, это ошибка, и вы можете отслеживать исправление здесь .

В зависимости от того, что вы передаете методу .train(), вы можете или не можете столкнуться с этой ошибкой.

0
Lauren 24 Окт 2019 в 19:27