File tree Expand file tree Collapse file tree 1 file changed +8
-9
lines changed
src/pypolymlp/calculator/thermodynamics Expand file tree Collapse file tree 1 file changed +8
-9
lines changed Original file line number Diff line number Diff 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 )
You can’t perform that action at this time.
0 commit comments