alex-pirozhenko / sklearn-pmml

A library that allows serialization of SciKit-Learn estimators into PMML
MIT License
70 stars 17 forks source link

Fixing GBRT, changing output format #20

Closed alex-pirozhenko closed 9 years ago

alex-pirozhenko commented 9 years ago

Hi all, I made a few changes to the way we output results from the resulting PMML model. First of all, now the varname outputs the label in the classification task, and varname::value outputs the probability to get value label. This allows us to deal with multiclass/multilabel models in future Also, the TransformationContext API was changed a bit, and all the schema names were extracted to the single enum.