From f39f6d5b0a907d519377e70876b32daad1a676f2 Mon Sep 17 00:00:00 2001
From: jvech <jmvalenciae@unal.edu.co>
Date: Tue, 3 Sep 2024 20:08:25 -0500
Subject: feat: shuffle dataset on each epoch done

The CLI option to disable it was also added.
---
 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 5dbb656..9f8e2a5 100644
--- a/src/nn.h
+++ b/src/nn.h
@@ -53,7 +53,8 @@ void nn_network_train(
         double *input, size_t input_shape[2],
         double *labels, size_t labels_shape[2],
         struct Cost cost, size_t epochs,
-        size_t batch_size, double alpha);
+        size_t batch_size, double alpha,
+        bool shuffle);
 
 void nn_layer_map_activation(
         double (*activation)(double),
-- 
cgit v1.2.3-70-g09d2