Added NGBClassifier and NGBRegressor to Encodable class hierarchy

Author: vruusmann

pmml-sklearn-extension/src/main/java/ngboost/NGBClassifier.java

@@ -27,6 +27,7 @@
 import org.dmg.pmml.MiningFunction;
 import org.dmg.pmml.Model;
 import org.dmg.pmml.OpType;
+import org.dmg.pmml.PMML;
 import org.dmg.pmml.mining.MiningModel;
 import org.dmg.pmml.mining.Segmentation.MissingPredictionTreatment;
 import org.dmg.pmml.regression.RegressionModel;
@@ -40,16 +41,25 @@
 import org.jpmml.converter.Schema;
 import org.jpmml.converter.mining.MiningModelUtil;
 import org.jpmml.converter.regression.RegressionModelUtil;
+import org.jpmml.sklearn.Encodable;
+import org.jpmml.sklearn.HasSkLearnOptions;
 import org.jpmml.sklearn.SkLearnException;
 import sklearn.Classifier;
+import sklearn.EstimatorUtil;
+import sklearn.HasFeatureNamesIn;
 import sklearn.Regressor;
 
-public class NGBClassifier extends Classifier {
+public class NGBClassifier extends Classifier implements HasFeatureNamesIn, HasSkLearnOptions, Encodable {
 
 	public NGBClassifier(String module, String name){
 		super(module, name);
 	}
 
+	@Override
+	public int getNumberOfFeatures(){
+		return getInteger("n_features");
+	}
+
 	@Override
 	public List<?> getClasses(){
 		Integer k = getK();
@@ -139,6 +149,11 @@ public MiningModel encodeCategoricalModel(Schema schema){
 		}
 	}
 
+	@Override
+	public PMML encodePMML(){
+		return EstimatorUtil.encodePMML(this);
+	}
+
 	public List<List<Regressor>> getBaseModels(){
 		return NGBoostUtil.getBaseModels(this);
 	}

pmml-sklearn-extension/src/main/java/ngboost/NGBRegressor.java

@@ -37,6 +37,7 @@
 import org.dmg.pmml.OpType;
 import org.dmg.pmml.Output;
 import org.dmg.pmml.OutputField;
+import org.dmg.pmml.PMML;
 import org.dmg.pmml.PMMLFunctions;
 import org.dmg.pmml.ParameterField;
 import org.dmg.pmml.ResultFeature;
@@ -52,16 +53,24 @@
 import org.jpmml.converter.mining.MiningModelUtil;
 import org.jpmml.python.ClassDictConstructorUtil;
 import org.jpmml.python.ClassDictUtil;
+import org.jpmml.sklearn.Encodable;
 import org.jpmml.sklearn.SkLearnException;
+import sklearn.EstimatorUtil;
+import sklearn.HasFeatureNamesIn;
 import sklearn.HasRegressorOptions;
 import sklearn.Regressor;
 
-public class NGBRegressor extends Regressor implements HasRegressorOptions {
+public class NGBRegressor extends Regressor implements HasFeatureNamesIn, HasRegressorOptions, Encodable {
 
 	public NGBRegressor(String module, String name){
 		super(module, name);
 	}
 
+	@Override
+	public int getNumberOfFeatures(){
+		return getInteger("n_features");
+	}
+
 	@Override
 	public MiningModel encodeModel(Schema schema){
 		String distName = getDistName();
@@ -198,6 +207,11 @@ public MiningModel encodePoissonModel(Schema schema){
 		return MiningModelUtil.createModelChain(Arrays.asList(locModel, regressionModel), Segmentation.MissingPredictionTreatment.RETURN_MISSING);
 	}
 
+	@Override
+	public PMML encodePMML(){
+		return EstimatorUtil.encodePMML(this);
+	}
+
 	public List<List<Regressor>> getBaseModels(){
 		return NGBoostUtil.getBaseModels(this);
 	}