Fixed the formatting of integer split conditions

jpmml · Mar 27, 2022 · 0460161 · 0460161
1 parent aa985bf
commit 0460161
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/pmml-xgboost/src/main/java/org/jpmml/xgboost/RegTree.java b/pmml-xgboost/src/main/java/org/jpmml/xgboost/RegTree.java
@@ -344,7 +344,15 @@ private org.dmg.pmml.tree.Node encodeNode(int index, Predicate predicate, boolea
 				DataType dataType = continuousFeature.getDataType();
 				switch(dataType){
 					case INTEGER:
-						splitValue = (int)(splitValue.floatValue() + 1f);
+						Float flooredSplitValue = (float)Math.floor(splitValue.floatValue());
+
+						if(splitValue.floatValue() == flooredSplitValue.floatValue()){
+							splitValue = (int)flooredSplitValue.floatValue();
+						} else
+
+						{
+							splitValue = (int)(flooredSplitValue.floatValue() + 1f);
+						}
 						break;
 					case FLOAT:
 						break;