From b715671988a4f3e476586df52fa3bf052cce7f80 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 05 Dec 2013 21:17:16 +0000
Subject: [PATCH] Works well on MNIST
---
src/data.h | 6 +++---
1 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/src/data.h b/src/data.h
index fbcb144..c01384c 100644
--- a/src/data.h
+++ b/src/data.h
@@ -9,9 +9,9 @@
double **truth;
} batch;
-batch get_all_data(char *filename);
-batch random_batch(char *filename, int n);
-batch get_batch(char *filename, int curr, int total);
+batch get_all_data(char *filename, char **labels, int k);
+batch random_batch(char *filename, int n, char **labels, int k);
+batch get_batch(char *filename, int curr, int total, char **labels, int k);
void free_batch(batch b);
--
Gitblit v1.10.0