Как сохранить график важности функции xgboost в файл из блокнота Jupyter

Question

Я борюсь за сохранение графика важности функций xgboost в файл. Я создал модель и наметил важность функций в своем блокноте Jupyter-

xgb_model = xgboost.train(best_params, dtrain, num_round)
xgboost.plot_importance(xgb_model)

Показывает график важности функций, но я не могу сохранить его в файл. Я даже искал атрибут save в dir(xgboost.plot_importance(xgb_model)), но ничего не получил. Есть ли способ сделать это?

Answer 1

Из документации вы видите, что это matplotlib вывод.Так что вы должны быть в состоянии вызвать savefig из matplotlib.

Если вы хотите сохранить модель, взгляните на Как сохранить и загрузить модель xgboost? .