Unable to export code from XGBoost 1.7.5 models about m2cgen HOT 6 OPEN

meet the same error

from m2cgen.

Debugging into m2cgen code, maybe I managed to find a workaround for my case (export a native Booster to C and Python code), but I do not know if it works for other cases.

If the base_score parameter of the classifier is forced to 0 (the default value in the latest version is None), the check is passed and both Python and C code are generated.

# booster is already trained
temp_file = "temp.ubj"
booster.save_model(temp_file)
xgbclf = xgb.XGBClassifier()
xgbclf.load_model(temp_file)
os.remove(temp_file)
xgbclf.base_score = 0  # workaround
c_code = m2c.export_to_c(xgbclf)
py_code = m2c.export_to_python(xgbclf)

from m2cgen.

Encountering the same error. This is a breaking change for my pipeline. Any ideas here?

from m2cgen.

Encountering the same error. This is a breaking change for my pipeline. Any ideas here?

Yep, it is breaking for me too, I hope for some updates from the developer...

from m2cgen.

Taking a look at m2cgen/assemblers/boosting.py (line 78:80), I think the problem depends on self._base_score being None for some reason in the latest XGBoost model version, so the check if self._base_score != 0.0 is passed and the next instruction fails base_score = -math.log(1.0 / self._base_score - 1.0).

Maybe this model field has been renamed or removed, I will check it out and update this issue.

from m2cgen.

Debugging into m2cgen code, maybe I managed to find a workaround for my case (export a native Booster to C and Python code), but I do not know if it works for other cases.

If the base_score parameter of the classifier is forced to 0 (the default value in the latest version is None), the check is passed and both Python and C code are generated.

# booster is already trained
temp_file = "temp.ubj"
booster.save_model(temp_file)
xgbclf = xgb.XGBClassifier()
xgbclf.load_model(temp_file)
os.remove(temp_file)
xgbclf.base_score = 0  # workaround
c_code = m2c.export_to_c(xgbclf)
py_code = m2c.export_to_python(xgbclf)

This works for me. Thanks!

from m2cgen.