From b2b7137b6f185ce2f01664d782a09b08d50d5a07 Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@gmail.com> Date: Tue, 28 Jan 2014 07:16:56 +0000 Subject: [PATCH] About to do something stupid... --- src/network.h | 6 ++++-- 1 files changed, 4 insertions(+), 2 deletions(-) diff --git a/src/network.h b/src/network.h index 3614c52..fa109dd 100644 --- a/src/network.h +++ b/src/network.h @@ -22,10 +22,12 @@ network make_network(int n); void forward_network(network net, double *input); -void backward_network(network net, double *input, double *truth); +double 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