Skip to content

Commit

Permalink
Merge pull request #314 from vikram-s-narayan/fix-neural-docs
Browse files Browse the repository at this point in the history
Fixes Issue - Docs for NeuralSurrogate #305
  • Loading branch information
ChrisRackauckas authored Jan 23, 2022
2 parents de9d919 + 3a9ad67 commit 8abf4ae
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions docs/src/neural.md
Original file line number Diff line number Diff line change
Expand Up @@ -52,15 +52,15 @@ As always, getting the model right is hardest thing.

```@example Neural_surrogate
model1 = Chain(
Dense(1, 5, σ),
Dense(2, 5, σ),
Dense(5,2,σ),
Dense(2, 1)
)
neural = NeuralSurrogate(x, y, lb, ub, model = model1, n_echos = 10)
neural = NeuralSurrogate(xys, zs, lower_bound, upper_bound, model = model1, n_echos = 10)
```

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

0 comments on commit 8abf4ae

Please sign in to comment.