Merge pull request #57 from SoftwareAG/ZAC-717

Added derived fields to segment's mining schema
SoftwareAG · May 5, 2022 · 3acb2b5 · 3acb2b5
2 parents e81389a + 50d75a5
commit 3acb2b5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/nyoka/skl/skl_to_pmml.py b/nyoka/skl/skl_to_pmml.py
@@ -1370,6 +1370,8 @@ def get_inner_segments(model, derived_col_names, col_names, index):
                 recordCount=estm.tree_.n_node_samples[0]
             )
         mining_fields = list()
+        if model.__class__.__name__ in ['RandomForestClassifier','RandomForestRegressor']:
+            col_names = derived_col_names
         for feat in col_names:
             mining_fields.append(pml.MiningField(name=feat))
         segments.append(