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.h | 4 +++- 1 files changed, 3 insertions(+), 1 deletions(-) diff --git a/src/network.h b/src/network.h index 3614c52..975c3dd 100644 --- a/src/network.h +++ b/src/network.h @@ -24,8 +24,10 @@ 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); +double train_network_batch(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