From 3a9ad6717344452f8d202e0606fcca6aff7c10aa Mon Sep 17 00:00:00 2001 From: Vikram Date: Sun, 23 Jan 2022 17:37:03 +0530 Subject: [PATCH] fix Docs for NeuralSurrogate #305 --- docs/src/neural.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/docs/src/neural.md b/docs/src/neural.md index 6b3c98816..551c8994e 100644 --- a/docs/src/neural.md +++ b/docs/src/neural.md @@ -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) ```