将变量重要性图导出为图像

【问题标题】：Exporting Variable Importance Plot as Image将变量重要性图导出为图像
【发布时间】：2019-11-30 22:30:21
【问题描述】：

我正在尝试将使用 H2O 库在 Jupyter 笔记本中生成的图像导出为 PNG 文件。图像是变量重要性图

我曾尝试使用 matplotlib 导出功能，但它会生成一个空的 PNG 文件。

cv_gbm.varimp_plot()

【问题讨论】：

标签： python-3.x h2o

【解决方案1】：

我不知道您的问题的直接解决方案，但以下步骤应该可行：

gbm_var_imp = cv_gbm._model_json['output']['variable_importances'].as_data_frame()

x = gbm_var_imp['scaled_importance']
x.index = gbm_var_imp['variable']

fig = plt.figure(figsize=(8, 8))
x.sort_values().plot(kind='barh')
fig.savefig('gbm_variable_importance.png', dpi=600)

【讨论】：