@@ -19,7 +19,7 @@ using SafeTestsets
1919 my_opt = Descent (0.01 )
2020 n_echos = 1
2121 my_neural = NeuralSurrogate (x, y, a, b, model = my_model, loss = my_loss, opt = my_opt,
22- n_echos = 1 )
22+ n_echos = 1 )
2323 my_neural_kwargs = NeuralSurrogate (x, y, a, b)
2424 add_point! (my_neural, 8.5 , 20.0 )
2525 add_point! (my_neural, [3.2 , 3.5 ], [7.4 , 8.0 ])
@@ -37,7 +37,7 @@ using SafeTestsets
3737 my_opt = Descent (0.01 )
3838 n_echos = 1
3939 my_neural = NeuralSurrogate (x, y, lb, ub, model = my_model, loss = my_loss,
40- opt = my_opt, n_echos = 1 )
40+ opt = my_opt, n_echos = 1 )
4141 my_neural_kwargs = NeuralSurrogate (x, y, lb, ub)
4242 my_neural ((3.5 , 1.49 ))
4343 my_neural ([3.4 , 1.4 ])
@@ -54,7 +54,7 @@ using SafeTestsets
5454 my_model = Chain (Dense (1 , 2 ))
5555 my_loss (x, y) = Flux. mse (my_model (x), y)
5656 surrogate = NeuralSurrogate (x, y, lb, ub, model = my_model, loss = my_loss,
57- opt = my_opt, n_echos = 1 )
57+ opt = my_opt, n_echos = 1 )
5858 surr_kwargs = NeuralSurrogate (x, y, lb, ub)
5959
6060 f = x -> [x[1 ], x[2 ]^ 2 ]
@@ -66,7 +66,7 @@ using SafeTestsets
6666 my_model = Chain (Dense (2 , 2 ))
6767 my_loss (x, y) = Flux. mse (my_model (x), y)
6868 surrogate = NeuralSurrogate (x, y, lb, ub, model = my_model, loss = my_loss,
69- opt = my_opt, n_echos = 1 )
69+ opt = my_opt, n_echos = 1 )
7070 surrogate_kwargs = NeuralSurrogate (x, y, lb, ub)
7171 surrogate ((1.0 , 2.0 ))
7272 x_new = (2.0 , 2.0 )
@@ -85,7 +85,7 @@ using SafeTestsets
8585 n_echos = 1
8686 my_neural_ND_neural = NeuralSurrogate (x, y, lb, ub)
8787 surrogate_optimize (objective_function_ND, SRBF (), lb, ub, my_neural_ND_neural,
88- SobolSample (), maxiters = 15 )
88+ SobolSample (), maxiters = 15 )
8989
9090 # AD Compatibility
9191 lb = 0.0
@@ -101,7 +101,7 @@ using SafeTestsets
101101 my_opt = Descent (0.01 )
102102 n_echos = 1
103103 my_neural = NeuralSurrogate (x, y, lb, ub, model = my_model, loss = my_loss,
104- opt = my_opt, n_echos = 1 )
104+ opt = my_opt, n_echos = 1 )
105105 g = x -> my_neural' (x)
106106 g (3.4 )
107107 end
@@ -120,7 +120,7 @@ using SafeTestsets
120120 my_opt = Descent (0.01 )
121121 n_echos = 1
122122 my_neural = NeuralSurrogate (x, y, lb, ub, model = my_model, loss = my_loss,
123- opt = my_opt, n_echos = 1 )
123+ opt = my_opt, n_echos = 1 )
124124 g = x -> Zygote. gradient (my_neural, x)
125125 g ((2.0 , 5.0 ))
126126 end
@@ -141,7 +141,7 @@ using SafeTestsets
141141 my_opt = Descent (0.01 )
142142 n_echos = 1
143143 my_neural = NeuralSurrogate (x, y, lb, ub, model = my_model, loss = my_loss,
144- opt = my_opt, n_echos = 1 )
144+ opt = my_opt, n_echos = 1 )
145145 Zygote. gradient (x -> sum (my_neural (x)), (2.0 , 5.0 ))
146146
147147 my_rad = RadialBasis (x, y, lb, ub, rad = linearRadial ())
0 commit comments