GH-15809: fix writing new parameter in MOJO load

h2oai · Feb 5, 2024 · 48939fd · 48939fd
1 parent ac8d8ab
commit 48939fd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/h2o-algos/src/main/java/hex/glm/GLMMojoWriter.java b/h2o-algos/src/main/java/hex/glm/GLMMojoWriter.java
@@ -40,7 +40,7 @@ protected void writeModelData() throws IOException {
     if (GLMModel.GLMParameters.Family.tweedie.equals(model._parms._family))
       writekv("tweedie_link_power", model._parms._tweedie_link_power);
 
-    writekv("dispersion_estimated", model._parms._compute_p_values ? model._parms._dispersion_estimated : 0);
+    writekv("dispersion_estimated", (model._parms._compute_p_values ? model._parms._dispersion_estimated : 1.0));
   }
 
 }