Actually add the argument this time

quantling · Sep 10, 2024 · 8832ed5 · 8832ed5
1 parent 4cefe48
commit 8832ed5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/deep_learning.jl b/src/deep_learning.jl
@@ -108,6 +108,7 @@ function get_and_train_model(X_train::Union{SparseMatrixCSC,Matrix},
                             verbose::Bool=true,
                             measures_func::Union{Missing, Function}=missing,
                             return_train_acc::Bool=false,
+                            regularisation_func::Union{Missing, Function} = missing,
                             kargs...)
 
     data_train = deepcopy(data_train)