Open phaidara opened 2 years ago
The JPMML-LightGBM library treats cross_entropy
as a classification-type objective function:
https://github.com/jpmml/jpmml-lightgbm/blob/1.4.2/pmml-lightgbm/src/main/java/org/jpmml/lightgbm/GBDT.java#L549-L553
It's currently unclear to me if it can and should be usable in regression contexts. Will explore.
Hello,
I am currently working on a project where I want to fit a model on probabilities and save it to PMML for later use in Java program.
I am training a LightGBMRegressor with the
cross_entropy
objective function. The training part is working well. I am able to fit a PMMLPipeline on my data and use it to predict probabilities as expected.But the saving to PMML part is failing with the following exception:
It seems that the cross_entropy objective function is not compatible with the LGBMRegressor in the jpmml LightGBM Java API. I tested the
cross_entropy
with an LGBMClassifier and binary targets (0, 1) instead of probabilities, and this is working fine.Would it be possible to fix this behavior? Thanks!
Reproducible example: