From aeb27a882087ee447a155bd589bd3712050a16f5 Mon Sep 17 00:00:00 2001
From: jvech <jmvalenciae@unal.edu.co>
Date: Sun, 6 Aug 2023 15:13:03 -0500
Subject: add: nn_network_train() added

The function needs to be tested
---
 src/nn.h | 11 +++++++++++
 1 file changed, 11 insertions(+)

(limited to 'src/nn.h')

diff --git a/src/nn.h b/src/nn.h
index 0c794b5..40066e3 100644
--- a/src/nn.h
+++ b/src/nn.h
@@ -10,6 +10,11 @@
 #include <unistd.h>
 #include <openblas/cblas.h>
 
+struct Cost {
+    double (*func)(double labels, double net_out);
+    double (*dfunc_out)(double labels, double net_out);
+};
+
 struct Activation {
     double (*func)(double);
     double (*dfunc)(double);
@@ -24,6 +29,12 @@ typedef struct Layer {
 void nn_network_init_weights(Layer *network, size_t nmemb, size_t input_cols);
 void nn_network_free_weights(Layer *network, size_t nmemb);
 
+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);
+
 void nn_layer_map_activation(
         double (*activation)(double),
         double *aout, size_t aout_shape[2],
-- 
cgit v1.2.3-70-g09d2