Improved support for binary and categorical features

jpmml · Feb 13, 2024 · f95cae2 · f95cae2
1 parent cceeb4f
commit f95cae2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pmml-lightgbm/src/main/java/org/jpmml/lightgbm/GBDT.java b/pmml-lightgbm/src/main/java/org/jpmml/lightgbm/GBDT.java
@@ -341,7 +341,7 @@ public Feature apply(Feature feature){
 					BinaryFeature binaryFeature = (BinaryFeature)feature;
 
 					Boolean binary = isBinary(index);
-					if(binary != null && binary.booleanValue()){
+					if(binary == null || binary.booleanValue()){
 						return binaryFeature;
 					}
 
@@ -357,7 +357,7 @@ public Feature apply(Feature feature){
 					CategoricalFeature categoricalFeature = (CategoricalFeature)feature;
 
 					Boolean categorical = isCategorical(index);
-					if(categorical != null && categorical.booleanValue()){
+					if(categorical == null || categorical.booleanValue()){
 						return categoricalFeature;
 					}
 				} else