From 8832ed558271d5091c0977399f76bb8c9ebda709 Mon Sep 17 00:00:00 2001
From: MariaHei <mabuwu83@gmail.com>
Date: Tue, 10 Sep 2024 18:22:13 +0100
Subject: [PATCH] Actually add the argument this time

---
 src/deep_learning.jl | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/deep_learning.jl b/src/deep_learning.jl
index 67ca751..f75d236 100644
--- 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)