From 1d53b6414e0cd81043d7c76aa89f4f97da5e479f Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Thu, 23 Jan 2014 19:24:37 +0000
Subject: [PATCH] Stable on MNIST, about to change a lot
---
src/utils.c | 84 ++++++++++++++++++++++++++++++++++++++++++
1 files changed, 84 insertions(+), 0 deletions(-)
diff --git a/src/utils.c b/src/utils.c
index 9848d08..5180fe6 100644
--- a/src/utils.c
+++ b/src/utils.c
@@ -143,5 +143,89 @@
return field;
}
+double mean_array(double *a, int n)
+{
+ int i;
+ double sum = 0;
+ for(i = 0; i < n; ++i) sum += a[i];
+ return sum/n;
+}
+double variance_array(double *a, int n)
+{
+ int i;
+ double sum = 0;
+ double mean = mean_array(a, n);
+ for(i = 0; i < n; ++i) sum += (a[i] - mean)*(a[i]-mean);
+ double variance = sum/n;
+ return variance;
+}
+
+double constrain(double a, double max)
+{
+ if(a > abs(max)) return abs(max);
+ if(a < -abs(max)) return -abs(max);
+ return a;
+}
+
+void normalize_array(double *a, int n)
+{
+ int i;
+ double mu = mean_array(a,n);
+ double sigma = sqrt(variance_array(a,n));
+ for(i = 0; i < n; ++i){
+ a[i] = (a[i] - mu)/sigma;
+ }
+ mu = mean_array(a,n);
+ sigma = sqrt(variance_array(a,n));
+}
+
+void translate_array(double *a, int n, double s)
+{
+ int i;
+ for(i = 0; i < n; ++i){
+ a[i] += s;
+ }
+}
+
+void scale_array(double *a, int n, double s)
+{
+ int i;
+ for(i = 0; i < n; ++i){
+ a[i] *= s;
+ }
+}
+int max_index(double *a, int n)
+{
+ if(n <= 0) return -1;
+ int i, max_i = 0;
+ double max = a[0];
+ for(i = 1; i < n; ++i){
+ if(a[i] > max){
+ max = a[i];
+ max_i = i;
+ }
+ }
+ return max_i;
+}
+
+double rand_normal()
+{
+ int i;
+ double sum= 0;
+ 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