我试图将以下XGBoost模型转换为CoreML文件但我收到错误 .
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "/usr/local/lib/python2.7/dist-
packages/coremltools/converters/xgboost/_tree.py", line 48, in convert
return _MLModel(_convert_tree_ensemble(model, feature_names, target))
File "/usr/local/lib/python2.7/dist-
packages/coremltools/converters/xgboost/_tree_ensemble.py", line 111, in convert_tree_ensemble
xgb_model_str = model.get_dump(with_stats=True, dump_format = 'json')
TypeError: get_dump() got an unexpected keyword argument 'dump_format'
我用pip安装了xgboost . 为什么会这样?
1 回答
按照安装指南 . 不要使用pip install xgboost,否则你会在get_dump中遇到这个错误 . 有关更多信息,请参见http://xgboost.readthedocs.io/en/latest/build.html .