Improved support for the 'ExplainableBoostingRegressor.link_' attribu…

…te. Fixes jpmml/sklearn2pmml#422
jpmml · Jun 19, 2024 · 1ac6e70 · 1ac6e70
1 parent 874e6e9
commit 1ac6e70
Show file tree

Hide file tree

Showing 4 changed files with 397 additions and 394 deletions.
diff --git a/...-sklearn-extension/src/main/java/interpret/glassbox/ebm/ExplainableBoostingRegressor.java b/...-sklearn-extension/src/main/java/interpret/glassbox/ebm/ExplainableBoostingRegressor.java
@@ -74,7 +74,7 @@ public Number getIntercept(){
 	}
 
 	public String getLink(){
-		return getEnum("link_", this::getString, Arrays.asList(ExplainableBoostingRegressor.LINK_IDENTITY));
+		return getEnum("link_", this::getString, Arrays.asList(ExplainableBoostingRegressor.LINK_IDENTITY, ExplainableBoostingRegressor.LINK_LOG));
 	}
 
 	@Override
@@ -93,10 +93,13 @@ private RegressionModel.NormalizationMethod parseLink(String link){
 		switch(link){
 			case ExplainableBoostingRegressor.LINK_IDENTITY:
 				return RegressionModel.NormalizationMethod.NONE;
+			case ExplainableBoostingRegressor.LINK_LOG:
+				return RegressionModel.NormalizationMethod.EXP;
 			default:
 				throw new IllegalArgumentException(link);
 		}
 	}
 
 	private static final String LINK_IDENTITY = "identity";
+	private static final String LINK_LOG = "log";
 }