fixed bug in last dimensionality fix

davidsebfischer · davidsebfischer · commit 6d848b22fb2f · 2020-02-25T17:29:14.000+01:00
diff --git a/batchglm/train/numpy/base_glm/estimator.py b/batchglm/train/numpy/base_glm/estimator.py
@@ -280,7 +280,7 @@ def iwls_step(
                     ).squeeze().T.compute()
                 elif len(idx_update[invertible]) == 1:
                     delta_theta[:, idx_update[invertible]] = np.expand_dims(
-                        np.linalg.solve(a[invertible], b[invertible]).compute(),
+                        np.linalg.solve(a[invertible[0]], b[invertible[0]]).compute(),
                         axis=-1
                     )
             else:

Original file line number	Diff line number	Diff line change
`@@ -280,7 +280,7 @@ def iwls_step(`
`280`	`280`	`).squeeze().T.compute()`
`281`	`281`	`elif len(idx_update[invertible]) == 1:`
`282`	`282`	`delta_theta[:, idx_update[invertible]] = np.expand_dims(`
`283`		`- np.linalg.solve(a[invertible], b[invertible]).compute(),`
	`283`	`+ np.linalg.solve(a[invertible[0]], b[invertible[0]]).compute(),`
`284`	`284`	`axis=-1`
`285`	`285`	`)`
`286`	`286`	`else:`