fixed test coeff normalization/standardization.

h2oai · Oct 11, 2024 · a537924 · a537924
1 parent 68f3a5e
commit a537924
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/h2o-py/tests/pyunit_utils/utils_for_glm_hglm_tests.py b/h2o-py/tests/pyunit_utils/utils_for_glm_hglm_tests.py
@@ -158,14 +158,13 @@ def normalize_coefs(coefs, numerical_cols, training_frame):
     """
     intercept_adjust = 0
     all_coefs_names = coefs.keys()
-    enum_cols = list(filter(lambda x: x not in numerical_cols, all_coefs_names))
     normalized_coefs = coefs.copy()
     # only numerical coefficients are changed.
     for cname in numerical_cols:
         cmean = training_frame[cname].mean()[0,0]
         csigma = training_frame[cname].sd()
-        normalized_coefs[cname] = coefs[cname] * sd
-        intercept_adjust = intercept_adjust + normalized_coefs[cname]*cmean/sd
+        normalized_coefs[cname] = coefs[cname] * csigma
+        intercept_adjust = intercept_adjust + normalized_coefs[cname]*cmean/csigma
     if "intercept" in all_coefs_names:
         normalized_coefs["intercept"] = coefs["intercept"]+intercept_adjust
     else: