Updated JPMML-XGBoost dependency

jpmml · Jun 20, 2024 · 85df79d · 85df79d
1 parent d660886
commit 85df79d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 7 deletions.
diff --git a/pmml-sparkml-xgboost/src/main/java/org/jpmml/sparkml/xgboost/BoosterUtil.java b/pmml-sparkml-xgboost/src/main/java/org/jpmml/sparkml/xgboost/BoosterUtil.java
@@ -69,16 +69,13 @@ public <M extends Model<M> & HasPredictionCol & GeneralParams, C extends ModelCo
 		Map<String, Object> options = new LinkedHashMap<>();
 		options.put(HasXGBoostOptions.OPTION_MISSING, converter.getOption(HasXGBoostOptions.OPTION_MISSING, missing));
 		options.put(HasXGBoostOptions.OPTION_COMPACT, converter.getOption(HasXGBoostOptions.OPTION_COMPACT, false));
+		options.put(HasXGBoostOptions.OPTION_INPUT_FLOAT, converter.getOption(HasXGBoostOptions.OPTION_INPUT_FLOAT, null));
 		options.put(HasXGBoostOptions.OPTION_NUMERIC, converter.getOption(HasXGBoostOptions.OPTION_NUMERIC, true));
 		options.put(HasXGBoostOptions.OPTION_PRUNE, converter.getOption(HasXGBoostOptions.OPTION_PRUNE, false));
 		options.put(HasXGBoostOptions.OPTION_NTREE_LIMIT, converter.getOption(HasXGBoostOptions.OPTION_NTREE_LIMIT, null));
 
-		Schema xgbSchema = learner.configureSchema(options, schema);
+		Schema xgbSchema = learner.toXGBoostSchema(schema);
 
-		MiningModel miningModel = learner.encodeModel(options, xgbSchema);
-
-		miningModel = learner.configureModel(options, miningModel);
-
-		return miningModel;
+		return learner.encodeModel(options, xgbSchema);
 	}
 }
diff --git a/pom.xml b/pom.xml
@@ -113,7 +113,7 @@
 			<dependency>
 				<groupId>org.jpmml</groupId>
 				<artifactId>pmml-xgboost</artifactId>
-				<version>1.8.4</version>
+				<version>1.8.5</version>
 			</dependency>
 
 			<dependency>