fix PolySpec regularization dimension mismatch (#296)

bd-j · Dec 11, 2023 · 6583fd0 · 6583fd0
1 parent dd50479
commit 6583fd0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/prospect/models/sedmodel.py b/prospect/models/sedmodel.py
@@ -689,7 +689,7 @@ def spec_calibration(self, theta=None, obs=None, spec=None, **kwargs):
             ATA = np.dot(A.T, A / yvar[:, None])
             reg = self.params.get('poly_regularization', 0.)
             if np.any(reg > 0):
-                ATA += reg**2 * np.eye(order)
+                ATA += reg**2 * np.eye(order+1)
             ATAinv = np.linalg.inv(ATA)
             c = np.dot(ATAinv, np.dot(A.T, y / yvar))
             Afull = chebvander(x, order)