Skip to content

Commit 62a7ca1

Browse files
committed
Implement extrapolation TI
1 parent 396627c commit 62a7ca1

File tree

1 file changed

+8
-9
lines changed

1 file changed

+8
-9
lines changed

src/pypolymlp/calculator/thermodynamics/fit_utils.py

Lines changed: 8 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -144,16 +144,15 @@ def _fit_single(
144144
if weight_begin or weight_end:
145145
weight_vals = np.ones(X.shape[0])
146146
if weight_begin:
147-
weight_vals[0] = 5.0
148-
weight_vals[1] = 4.0
149-
weight_vals[2] = 3.0
150-
weight_vals[3] = 2.0
147+
weight_vals[0] = 5.0**2
148+
weight_vals[1] = 4.0**2
149+
weight_vals[2] = 3.0**2
150+
weight_vals[3] = 2.0**2
151151
if weight_end:
152-
weight_vals[-1] = 5.0
153-
weight_vals[-2] = 4.0
154-
weight_vals[-3] = 3.0
155-
weight_vals[-4] = 2.0
156-
152+
weight_vals[-1] = 5.0**2
153+
weight_vals[-2] = 4.0**2
154+
weight_vals[-3] = 3.0**2
155+
weight_vals[-4] = 2.0**2
157156
W = np.diag(weight_vals)
158157
try:
159158
coeffs = np.linalg.solve(X.T @ W @ X, X.T @ W @ y)

0 commit comments

Comments
 (0)