From e8624e1ebcabcc831d651e0beefe32df1463c903 Mon Sep 17 00:00:00 2001 From: jvech Date: Mon, 26 Aug 2024 12:12:40 -0500 Subject: add: mini batch learning implemented --- src/nn.h | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'src/nn.h') diff --git a/src/nn.h b/src/nn.h index 13e81da..5dbb656 100644 --- a/src/nn.h +++ b/src/nn.h @@ -52,7 +52,8 @@ void nn_network_train( Layer network[], size_t network_size, double *input, size_t input_shape[2], double *labels, size_t labels_shape[2], - struct Cost cost, size_t epochs, double alpha); + struct Cost cost, size_t epochs, + size_t batch_size, double alpha); void nn_layer_map_activation( double (*activation)(double), -- cgit v1.2.3-70-g09d2