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 910e73e commit 5357883Copy full SHA for 5357883
examples/2-deep-kernel-learning/script.jl
@@ -54,9 +54,9 @@ rng = Random.default_rng()
54
ps, st = Lux.setup(rng, neuralnet)
55
56
# Create a wrapper function for the neural network that will be updated during training
57
-function neural_transform(x, params)
58
- # neuralnet returns (output, new_state), we only need the output
59
- return first(neuralnet(x, params, st))
+function neural_transform(x, θ)
+ # neuralnet returns (output, new_state), we only need the output
+ return first(neuralnet(x, θ, st))
60
end
61
62
# We use the Squared Exponential Kernel:
0 commit comments