From ac2ec745a828cf6498f83cf737eced246ed1e882 Mon Sep 17 00:00:00 2001 From: Villu Ruusmann Date: Mon, 22 Aug 2022 10:33:54 +0300 Subject: [PATCH] Updated JPMML-Converter dependency --- .../src/main/java/org/jpmml/xgboost/ObjFunction.java | 2 +- pom.xml | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/pmml-xgboost/src/main/java/org/jpmml/xgboost/ObjFunction.java b/pmml-xgboost/src/main/java/org/jpmml/xgboost/ObjFunction.java index 2bea1e1..7b9ba60 100644 --- a/pmml-xgboost/src/main/java/org/jpmml/xgboost/ObjFunction.java +++ b/pmml-xgboost/src/main/java/org/jpmml/xgboost/ObjFunction.java @@ -156,7 +156,7 @@ protected MiningModel createMiningModel(List trees, List weights MiningModel miningModel = new MiningModel(MiningFunction.REGRESSION, ModelUtil.createMiningSchema(continuousLabel)) .setMathContext(MathContext.FLOAT) - .setSegmentation(MiningModelUtil.createSegmentation(equalWeights ? Segmentation.MultipleModelMethod.SUM : Segmentation.MultipleModelMethod.WEIGHTED_SUM, treeModels, weights)) + .setSegmentation(MiningModelUtil.createSegmentation(equalWeights ? Segmentation.MultipleModelMethod.SUM : Segmentation.MultipleModelMethod.WEIGHTED_SUM, Segmentation.MissingPredictionTreatment.RETURN_MISSING, treeModels, weights)) .setTargets(ModelUtil.createRescaleTargets(null, intercept, continuousLabel)); return miningModel; diff --git a/pom.xml b/pom.xml index eeb9e98..04cc528 100644 --- a/pom.xml +++ b/pom.xml @@ -64,18 +64,18 @@ org.jpmml pmml-converter - 1.5.3 + 1.5.4 org.jpmml pmml-converter-testing - 1.5.3 + 1.5.4 org.jpmml pmml-evaluator-testing - 1.6.3 + 1.6.4