Сохранение модели xgboost в hdfs с использованием python

Question

Мне нужно сохранить модель XGBoost в hdfs, используя python. Я пытался сделать это, используя xgbm.save_model(path_to_hdfs) но я получил следующую ошибку: src/io.cc:37: Please compile with DMLC_USE_HDFS=1 to use hdfs

Итак, мой вопрос, как бороться с этой ошибкой? Или, может быть, есть другие способы хранения моделей XGBoost в формате hdf с помощью скрипта python?

Answer 1

Как говорит ошибка, вам нужно скомпилировать xgboost с поддержкой HDFS.

Для этого вам нужно сделать что-то вроде:

git clone --recursive https://github.com/dmlc/xgboost.git
cd xgboost
mkdir build
cmake -DDMLC_USE_HDFS=1 ..
make

Установите пакет Python и повторите попытку.