From 79fffcce3ce495bd415dc1284224c915d7194d4c Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 11 Dec 2014 21:15:26 +0000
Subject: [PATCH] Better imagenet distributed training

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

diff --git a/src/matrix.h b/src/matrix.h
index 01d825d..d84431c 100644
--- a/src/matrix.h
+++ b/src/matrix.h
@@ -11,7 +11,7 @@
 
 matrix csv_to_matrix(char *filename);
 matrix hold_out_matrix(matrix *m, int n);
-float matrix_accuracy(matrix truth, matrix guess);
+float matrix_topk_accuracy(matrix truth, matrix guess, int k);
 void matrix_add_matrix(matrix from, matrix to);
 
 float *pop_column(matrix *m, int c);

--
Gitblit v1.10.0