search

jpmml / jpmml-sklearn

Java library and command-line application for converting Scikit-Learn pipelines to PMML
GNU Affero General Public License v3.0
531 stars 117 forks source link

Add support for `FeatureUnion.transformer_weights` attribute #63

Open vruusmann opened 6 years ago

vruusmann commented 6 years ago

Somehow this attribute has gone unnoticed (maybe a recent addition?).

Two-step plan:

  1. The converter should be(come) aware of it, and fail with an IllegalArgumentException("Attribute FeatureUnion.transformer_weights is not supported") if it is initialized with non-null/non-unit weights. This is to prevent irreproducible transformation results between Scikit-Learn and (J)PMML.
  2. Add full support.