我在尝试将xgboost特征重要性图保存到文件中遇到了困难。我已经在jupyter notebook中创建了一个模型并绘制了特征的重要性-
xgb_model = xgboost.train(best_params, dtrain, num_round)
xgboost.plot_importance(xgb_model)
它展示了我特征重要性图,但是我不能将其保存到文件中。我甚至在dir(xgboost.plot_importance(xgb_model))
中寻找任何save属性,但没有得到任何东西。有没有什么方法可以做到这一点?