From ad9dbfe16495204453b1b7f8593d320751f76ca0 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Tue, 10 Dec 2013 18:30:42 +0000
Subject: [PATCH] CSE546 submission

---
 src/network.h |    3 ++-
 1 files changed, 2 insertions(+), 1 deletions(-)

diff --git a/src/network.h b/src/network.h
index 3614c52..2ffc76b 100644
--- a/src/network.h
+++ b/src/network.h
@@ -24,8 +24,9 @@
 void forward_network(network net, double *input);
 void backward_network(network net, double *input, double *truth);
 void update_network(network net, double step, double momentum, double decay);
-double train_network_sgd(network net, data d, double step, double momentum,double decay);
+double train_network_sgd(network net, data d, int n, double step, double momentum,double decay);
 void train_network(network net, data d, double step, double momentum, double decay);
+matrix network_predict_data(network net, data test);
 double network_accuracy(network net, data d);
 double *get_network_output(network net);
 double *get_network_output_layer(network net, int i);

--
Gitblit v1.10.0