Refactored ObjectiveFunction class

jpmml · Jul 14, 2024 · cfed33f · cfed33f
1 parent 6de0be4
commit cfed33f
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/pmml-lightgbm/src/main/java/org/jpmml/lightgbm/Classification.java b/pmml-lightgbm/src/main/java/org/jpmml/lightgbm/Classification.java
@@ -26,7 +26,7 @@
 import org.jpmml.converter.CategoricalLabel;
 import org.jpmml.converter.Label;
 import org.jpmml.converter.LabelUtil;
-import org.jpmml.converter.PMMLEncoder;
+import org.jpmml.converter.ModelEncoder;
 
 abstract
 public class Classification extends ObjectiveFunction {
@@ -41,7 +41,7 @@ public Classification(Section config){
 	}
 
 	@Override
-	public Label encodeLabel(String targetName, List<?> targetCategories, PMMLEncoder encoder){
+	public Label encodeLabel(String targetName, List<?> targetCategories, ModelEncoder encoder){
 		DataField dataField;
 
 		if(targetCategories == null){

diff --git a/pmml-lightgbm/src/main/java/org/jpmml/lightgbm/ObjectiveFunction.java b/pmml-lightgbm/src/main/java/org/jpmml/lightgbm/ObjectiveFunction.java
@@ -27,8 +27,8 @@
 import org.dmg.pmml.tree.TreeModel;
 import org.jpmml.converter.ContinuousLabel;
 import org.jpmml.converter.Label;
+import org.jpmml.converter.ModelEncoder;
 import org.jpmml.converter.ModelUtil;
-import org.jpmml.converter.PMMLEncoder;
 import org.jpmml.converter.PredicateManager;
 import org.jpmml.converter.Schema;
 import org.jpmml.converter.mining.MiningModelUtil;
@@ -47,7 +47,7 @@ public ObjectiveFunction(Section config){
 	}
 
 	abstract
-	public Label encodeLabel(String targetName, List<?> targetCategories, PMMLEncoder encoder);
+	public Label encodeLabel(String targetName, List<?> targetCategories, ModelEncoder encoder);
 
 	abstract
 	public MiningModel encodeModel(List<Tree> trees, Integer numIteration, Schema schema);

diff --git a/pmml-lightgbm/src/main/java/org/jpmml/lightgbm/Regression.java b/pmml-lightgbm/src/main/java/org/jpmml/lightgbm/Regression.java
@@ -26,7 +26,7 @@
 import org.dmg.pmml.mining.MiningModel;
 import org.jpmml.converter.ContinuousLabel;
 import org.jpmml.converter.Label;
-import org.jpmml.converter.PMMLEncoder;
+import org.jpmml.converter.ModelEncoder;
 import org.jpmml.converter.Schema;
 
 public class Regression extends ObjectiveFunction {
@@ -36,7 +36,7 @@ public Regression(Section config){
 	}
 
 	@Override
-	public Label encodeLabel(String targetName, List<?> targetCategories, PMMLEncoder encoder){
+	public Label encodeLabel(String targetName, List<?> targetCategories, ModelEncoder encoder){
 
 		if(targetCategories != null && targetCategories.size() > 0){
 			throw new IllegalArgumentException("Regression requires zero target categories");