From 14303717dcddae43cdc55beb0685dae86f566fd8 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Sat, 25 Oct 2014 18:57:26 +0000
Subject: [PATCH] Fast, needs to be faster

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

diff --git a/src/network.h b/src/network.h
index 22e277c..c95f6fa 100644
--- a/src/network.h
+++ b/src/network.h
@@ -52,8 +52,10 @@
 float train_network_batch(network net, data d, int n);
 void train_network(network net, data d);
 matrix network_predict_data(network net, data test);
+float *network_predict(network net, float *input);
 float network_accuracy(network net, data d);
 float network_accuracy_multi(network net, data d, int n);
+void top_predictions(network net, int n, int *index);
 float *get_network_output(network net);
 float *get_network_output_layer(network net, int i);
 float *get_network_delta_layer(network net, int i);

--
Gitblit v1.10.0