autodeployai / pmml4s

PMML scoring library for Scala
https://www.pmml4s.org/
Apache License 2.0
62 stars 10 forks source link

The LightGBM model output is quite different to the expected result. #26

Open shipengAlan opened 1 year ago

shipengAlan commented 1 year ago

I exported my tree based regression model to PMML format. And based on this library I load the model file, evaluate it with unit cases and found its output score is quite different to the result when training. Do you know the possible reasons? Thanks a lot

scorebot commented 1 year ago

About the results of the model prediction, you can refer to the third part Understand the result values of https://github.com/autodeployai/pmml4s#use-in-scala

There are some models that could define output fields without the predicted value, for example only probabilities of each category, then if you expect all the possible results, you can call the method model.setSupplementOutput(true) to get all results including the final predicted value.

scorebot commented 1 year ago

@shipengAlan Please let me know if you still have a problem