We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c5cae95 commit c04adbdCopy full SHA for c04adbd
lib/SurrogatesFlux/test/runtests.jl
@@ -53,6 +53,7 @@ using SafeTestsets
53
x = sample(20, lb, ub, SobolSample())
54
y = f.(x)
55
my_model = Chain(Dense(2, 2))
56
+ my_opt = Descent(0.01)
57
surrogate = NeuralSurrogate(x, y, lb, ub, model = my_model, loss = Flux.mse,
58
opt = my_opt, n_epochs = 1)
59
surrogate([1.0, 2.0])
0 commit comments