|
@@ -111,6 +111,8 @@ function adapt_learnrate(epoch_idx)
|
|
|
return learning_rate * decay_rate^(epoch_idx / decay_step)
|
|
|
end
|
|
|
|
|
|
+# TODO different idea for the accuracy: draw circle around ground truth and if prediction lays within the circle count this as a hit
|
|
|
+
|
|
|
function accuracy(model, x, y)
|
|
|
y_hat = Tracker.data(model(x))
|
|
|
return mean(mapslices(button_number, y_hat, dims=1) .== mapslices(button_number, y, dims=1))
|