Updated JPMML-Converter dependency

jpmml · May 14, 2019 · fd8626a · fd8626a
1 parent 8e76ab6
commit fd8626a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/pom.xml b/pom.xml
@@ -72,13 +72,13 @@
 		<dependency>
 			<groupId>org.jpmml</groupId>
 			<artifactId>jpmml-converter</artifactId>
-			<version>1.3.6</version>
+			<version>1.3.7</version>
 		</dependency>
 
 		<dependency>
 			<groupId>org.jpmml</groupId>
 			<artifactId>jpmml-xgboost</artifactId>
-			<version>1.3.6</version>
+			<version>1.3.8</version>
 		</dependency>
 
 		<dependency>
@@ -91,13 +91,13 @@
 		<dependency>
 			<groupId>org.jpmml</groupId>
 			<artifactId>pmml-evaluator</artifactId>
-			<version>1.4.8</version>
+			<version>1.4.9</version>
 			<scope>test</scope>
 		</dependency>
 		<dependency>
 			<groupId>org.jpmml</groupId>
 			<artifactId>pmml-evaluator-test</artifactId>
-			<version>1.4.8</version>
+			<version>1.4.9</version>
 			<scope>test</scope>
 		</dependency>
 	</dependencies>

diff --git a/src/main/java/org/jpmml/h2o/GbmMojoModelConverter.java b/src/main/java/org/jpmml/h2o/GbmMojoModelConverter.java
@@ -75,7 +75,7 @@ public MiningModel encodeModel(Schema schema){
 
 			MiningModel miningModel = new MiningModel(MiningFunction.REGRESSION, ModelUtil.createMiningSchema(continuousLabel))
 				.setSegmentation(MiningModelUtil.createSegmentation(MultipleModelMethod.SUM, treeModels))
-				.setTargets(ModelUtil.createRescaleTargets(null, (double)model._init_f, continuousLabel));
+				.setTargets(ModelUtil.createRescaleTargets(null, model._init_f, continuousLabel));
 
 			return miningModel;
 		} else
@@ -85,7 +85,7 @@ public MiningModel encodeModel(Schema schema){
 
 			MiningModel miningModel = new MiningModel(MiningFunction.REGRESSION, ModelUtil.createMiningSchema(continuousLabel))
 				.setSegmentation(MiningModelUtil.createSegmentation(MultipleModelMethod.SUM, treeModels))
-				.setTargets(ModelUtil.createRescaleTargets(null, (double)model._init_f, continuousLabel))
+				.setTargets(ModelUtil.createRescaleTargets(null, model._init_f, continuousLabel))
 				.setOutput(ModelUtil.createPredictedOutput(FieldName.create("gbmValue"), OpType.CONTINUOUS, DataType.DOUBLE));
 
 			return MiningModelUtil.createRegression(miningModel, RegressionModel.NormalizationMethod.EXP, schema);
@@ -96,7 +96,7 @@ public MiningModel encodeModel(Schema schema){
 
 			MiningModel miningModel = new MiningModel(MiningFunction.REGRESSION, ModelUtil.createMiningSchema(continuousLabel))
 				.setSegmentation(MiningModelUtil.createSegmentation(Segmentation.MultipleModelMethod.SUM, treeModels))
-				.setTargets(ModelUtil.createRescaleTargets(null, (double)model._init_f, continuousLabel))
+				.setTargets(ModelUtil.createRescaleTargets(null, model._init_f, continuousLabel))
 				.setOutput(ModelUtil.createPredictedOutput(FieldName.create("gbmValue"), OpType.CONTINUOUS, DataType.DOUBLE));
 
 			return MiningModelUtil.createBinaryLogisticClassification(miningModel, 1d, 0d, RegressionModel.NormalizationMethod.LOGIT, true, schema);