@@ -110,17 +110,14 @@ end
110110 component_pdf (x:: Real , mu:: Real , tau:: Real ) =
111111 tau * exp (- tau^ 2 * (x - mu)^ 2 / 2 ) / sqrt (2 π)
112112 data = randn (StableRNG (123 ), 40 )
113- loss (params:: AbstractVector ) = - sum (log,
114- (1 - params[1 ]) * component_pdf (x, 0 , 1 )
115- + params[1 ] * component_pdf (x, params[2 ], params[3 ])
116- for x in data
117- )
113+ loss (params:: AbstractVector ) =
114+ - sum (
115+ log,
116+ (1 - params[1 ]) * component_pdf (x, 0 , 1 ) +
117+ params[1 ] * component_pdf (x, params[2 ], params[3 ]) for x in data
118+ )
118119 model = Model (loss)
119- addvar! (model, [0 , - Inf , 0 ], [1 , Inf , Inf ], init= [0.1 , 0 , 1 ])
120- res = optimize (model, AugLag (), [0.1 , - 3 , 1 ], options= AugLagOptions ())
121- @test res. minimizer == [
122- 0.0 ,
123- - 2.3140286844577305 ,
124- 0.06904076799516767 ,
125- ]
120+ addvar! (model, [0 , - Inf , 0 ], [1 , Inf , Inf ], init = [0.1 , 0 , 1 ])
121+ res = optimize (model, AugLag (), [0.1 , - 3 , 1 ], options = AugLagOptions ())
122+ @test res. minimizer == [0.0 , - 2.3140286844577305 , 0.06904076799516767 ]
126123end
0 commit comments