Conversion du modèle XGBoost en CoreML

Question

J'essaie de convertir le modèle XGBoost suivant en un fichier CoreML, mais j'obtiens l'erreur.

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' 

J'ai installé xgboost en utilisant pip. Pourquoi cela se produit-il?

Answer 1

Suivez le guide d'installation. Ne pas utiliser pip installer xgboost ou vous obtiendrez cette erreur avec get_dump. Voir http://xgboost.readthedocs.io/en/latest/build.html pour plus d'informations.