From 8c3694bc911bbeab63e75c18f920e0991a5fa877 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Sat, 07 Dec 2013 17:38:50 +0000
Subject: [PATCH] Ensemble

---
 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