change T matrix initlization to be diagonal matrix

h2oai · Oct 27, 2024 · e877f21 · e877f21
1 parent 7a3a14f
commit e877f21
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/h2o-algos/src/test/java/hex/hglm/HGLMBasicTest.java b/h2o-algos/src/test/java/hex/hglm/HGLMBasicTest.java
@@ -583,14 +583,14 @@ public void testSetInitBetasTvar() {
       Frame tMat = new TestFrameBuilder()
               .withColNames("C1","C2","C3","C4","C5","C6","C7","C8")
               .withVecTypes(T_NUM, T_NUM, T_NUM, T_NUM, T_NUM, T_NUM, T_NUM, T_NUM)
-              .withDataForCol(0, new double[]{1.0173, 0.0009, 0.0009, 0.0004, 0.0001, 0.0001, 0.0008, 0.0000})
-              .withDataForCol(1, new double[]{0.0009, 1.2488, 0.0070, 0.0000, 0.0004, 0.0008, 0.0004, 0.0006})
-              .withDataForCol(2, new double[]{0.0009, 0.0070, 1.4728, 0.0001, 0.0006, 0.0006, 0.0007, 0.0000})
-              .withDataForCol(3, new double[]{0.0004, 0.0000, 0.0001, 1.4303, 0.0002, 0.0002, 0.0007, 0.0009})
-              .withDataForCol(4, new double[]{0.0001, 0.0004, 0.0006, 0.0002, 1.2905, 0.0003, 0.0002, 0.0002})
-              .withDataForCol(5, new double[]{0.0001, 0.0008, 0.0006, 0.0002, 0.0003, 1.7323, 0.0002, 0.0005})
-              .withDataForCol(6, new double[]{0.0008, 0.0004, 0.0007, 0.0007, 0.0002, 0.0002, 1.1840, 0.0007})
-              .withDataForCol(7, new double[]{0.0000, 0.0006, 0.0000, 0.0009, 0.0002, 0.0005, 0.0007, 1.3962})
+              .withDataForCol(0, new double[]{1.0173, 0, 0, 0, 0, 0, 0, 0})
+              .withDataForCol(1, new double[]{0, 1.2488, 0, 0, 0, 0, 0, 0})
+              .withDataForCol(2, new double[]{0, 0, 1.4728, 0, 0, 0, 0, 0})
+              .withDataForCol(3, new double[]{0, 0, 0, 1.4303, 0, 0, 0, 0})
+              .withDataForCol(4, new double[]{0, 0, 0, 0, 1.2905, 0, 0, 0})
+              .withDataForCol(5, new double[]{0, 0, 0, 0, 0, 1.7323, 0, 0})
+              .withDataForCol(6, new double[]{0, 0, 0, 0, 0, 0, 1.1840, 0})
+              .withDataForCol(7, new double[]{0, 0, 0, 0, 0, 0, 0, 1.3962})
               .build();
       Scope.track(tMat);
       Scope.track(prostate);