From 1d53b6414e0cd81043d7c76aa89f4f97da5e479f Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 23 Jan 2014 19:24:37 +0000
Subject: [PATCH] Stable on MNIST, about to change a lot
---
src/network.c | 18 ++++++++++++++++++
1 files changed, 18 insertions(+), 0 deletions(-)
diff --git a/src/network.c b/src/network.c
index 10ad110..07ac621 100644
--- a/src/network.c
+++ b/src/network.c
@@ -187,6 +187,24 @@
}
return (double)correct/n;
}
+double train_network_batch(network net, data d, int n, double step, double momentum,double decay)
+{
+ int i;
+ int correct = 0;
+ for(i = 0; i < n; ++i){
+ int index = rand()%d.X.rows;
+ double *x = d.X.vals[index];
+ double *y = d.y.vals[index];
+ forward_network(net, x);
+ int class = get_predicted_class_network(net);
+ backward_network(net, x, y);
+ correct += (y[class]?1:0);
+ }
+ update_network(net, step, momentum, decay);
+ return (double)correct/n;
+
+}
+
void train_network(network net, data d, double step, double momentum, double decay)
{
--
Gitblit v1.10.0