From 4bdf96bd6aafbec6bc3f0eab8739d6652878fd24 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Fri, 06 Dec 2013 21:26:09 +0000
Subject: [PATCH] New data format
---
src/utils.c | 13 +++++++++++++
1 files changed, 13 insertions(+), 0 deletions(-)
diff --git a/src/utils.c b/src/utils.c
index 3b8b5a8..5180fe6 100644
--- a/src/utils.c
+++ b/src/utils.c
@@ -216,3 +216,16 @@
for(i = 0; i < 12; ++i) sum += (double)rand()/RAND_MAX;
return sum-6.;
}
+
+double **one_hot_encode(double *a, int n, int k)
+{
+ int i;
+ double **t = calloc(n, sizeof(double*));
+ for(i = 0; i < n; ++i){
+ t[i] = calloc(k, sizeof(double));
+ int index = (int)a[i];
+ t[i][index] = 1;
+ }
+ return t;
+}
+
--
Gitblit v1.10.0