From 3a45e5ce7fbb95a82572c6d1a096a2a9d6e8bfc3 Mon Sep 17 00:00:00 2001 From: syzonyuliia Date: Fri, 26 Jan 2024 17:06:45 +0100 Subject: [PATCH] GH-15809: add new parameter to the constructor, and add new constructor --- h2o-algos/src/main/java/hex/glm/GLMModel.java | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/h2o-algos/src/main/java/hex/glm/GLMModel.java b/h2o-algos/src/main/java/hex/glm/GLMModel.java index 361759e91008..f4b71c1a1809 100755 --- a/h2o-algos/src/main/java/hex/glm/GLMModel.java +++ b/h2o-algos/src/main/java/hex/glm/GLMModel.java @@ -727,6 +727,11 @@ public GLMParameters(Family f, Link l, double [] lambda, double [] alpha, double public GLMParameters(Family f, Link l, double [] lambda, double [] alpha, double twVar, double twLnk, String[] interactions, double theta){ + this(f,l,lambda,alpha,twVar,twLnk,interactions, theta, Double.NaN); + } + + public GLMParameters(Family f, Link l, double [] lambda, double [] alpha, double twVar, double twLnk, + String[] interactions, double theta, double dispersion_estimated){ this._lambda = lambda; this._alpha = alpha; this._tweedie_variance_power = twVar; @@ -736,7 +741,7 @@ public GLMParameters(Family f, Link l, double [] lambda, double [] alpha, double _link = l; this._theta=theta; this._invTheta = 1.0/theta; - this._dispersion_estimated = _init_dispersion_parameter; + this._dispersion_estimated = Double.isNaN(dispersion_estimated) ? _init_dispersion_parameter : dispersion_estimated; } public final double variance(double mu){