aboutsummaryrefslogtreecommitdiff
path: root/src/nn.h
diff options
context:
space:
mode:
Diffstat (limited to 'src/nn.h')
-rw-r--r--src/nn.h4
1 files changed, 2 insertions, 2 deletions
diff --git a/src/nn.h b/src/nn.h
index 05ddd28..d66fabc 100644
--- a/src/nn.h
+++ b/src/nn.h
@@ -15,8 +15,8 @@ typedef struct Layer {
size_t neurons, input_nodes;
} Layer;
-void nn_layer_init_weights(Layer *layers, size_t nmemb, size_t input_cols);
-void nn_layer_free_weights(Layer *layer, size_t nmemb);
+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_layer_forward(Layer layer, double *out, size_t out_shape[2], double *input, size_t input_shape[2]); //TODO
void nn_layer_backward(Layer *layer, double *out, size_t out_shape[2]); //TODO
Feel free to download, copy and edit any repo