Updated JPMML-XGBoost dependency

See commit 2b86b14
jpmml · Apr 30, 2024 · 7f4a7f3 · 7f4a7f3
1 parent 11daeaa
commit 7f4a7f3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/pmml-h2o/src/main/java/org/jpmml/h2o/XGBoostMojoModelConverter.java b/pmml-h2o/src/main/java/org/jpmml/h2o/XGBoostMojoModelConverter.java
@@ -99,12 +99,8 @@ public MiningModel encodeModel(Schema schema){
 		options.put(HasXGBoostOptions.OPTION_COMPACT, Boolean.TRUE);
 		options.put(HasXGBoostOptions.OPTION_NUMERIC, Boolean.TRUE);
 
-		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
@@ -86,7 +86,7 @@
 			<dependency>
 				<groupId>org.jpmml</groupId>
 				<artifactId>pmml-xgboost</artifactId>
-				<version>1.8.4</version>
+				<version>1.8.5</version>
 			</dependency>
 
 			<dependency>