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/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