Skip to content

Commit 8abf4ae

Browse files
Merge pull request #314 from vikram-s-narayan/fix-neural-docs
Fixes Issue - Docs for NeuralSurrogate #305
2 parents de9d919 + 3a9ad67 commit 8abf4ae

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

docs/src/neural.md

+3-3
Original file line numberDiff line numberDiff line change
@@ -52,15 +52,15 @@ As always, getting the model right is hardest thing.
5252

5353
```@example Neural_surrogate
5454
model1 = Chain(
55-
Dense(1, 5, σ),
55+
Dense(2, 5, σ),
5656
Dense(5,2,σ),
5757
Dense(2, 1)
5858
)
59-
neural = NeuralSurrogate(x, y, lb, ub, model = model1, n_echos = 10)
59+
neural = NeuralSurrogate(xys, zs, lower_bound, upper_bound, model = model1, n_echos = 10)
6060
```
6161

6262
## Optimization
6363
We can now call an optimization function on the neural network:
6464
```@example Neural_surrogate
65-
surrogate_optimize(schaffer, SRBF(), lower_bound, upper_bound, neura, SobolSample(), maxiters=20, num_new_samples=10)
65+
surrogate_optimize(schaffer, SRBF(), lower_bound, upper_bound, neural, SobolSample(), maxiters=20, num_new_samples=10)
6666
```

0 commit comments

Comments
 (0)