1 files changed, 1 insertions, 1 deletions
diff --git a/src/nn.h b/src/nn.h
index 40066e3..2fcf9be 100644
--- a/src/nn.h
+++ b/src/nn.h
@@ -11,7 +11,7 @@
 #include <openblas/cblas.h>
 
 struct Cost {
-    double (*func)(double labels, double net_out);
+    double (*func)(double labels[], double net_out[], size_t shape);
     double (*dfunc_out)(double labels, double net_out);
 };