Python 3.x 将可变重要性图导出为图像

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

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

cv_gbm.varimp_plot()

---

我不知道你的问题的直接解决方案，但以下步骤应该有效：

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)

---

我不知道你的问题的直接解决方案，但以下步骤应该有效：

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)