From f7a17f82eb43de864a4f980f235055da9685eef8 Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Wed, 29 Jan 2014 00:28:42 +0000
Subject: [PATCH] Convolutional layers working w/ matrices
---
src/data.c | 20 ++++++++++----------
1 files changed, 10 insertions(+), 10 deletions(-)
diff --git a/src/data.c b/src/data.c
index 0b396d7..2c5932b 100644
--- a/src/data.c
+++ b/src/data.c
@@ -19,10 +19,10 @@
return lines;
}
-void fill_truth(char *path, char **labels, int k, double *truth)
+void fill_truth(char *path, char **labels, int k, float *truth)
{
int i;
- memset(truth, 0, k*sizeof(double));
+ memset(truth, 0, k*sizeof(float));
for(i = 0; i < k; ++i){
if(strstr(path, labels[i])){
truth[i] = 1;
@@ -36,7 +36,7 @@
data d;
d.shallow = 0;
d.X.rows = n;
- d.X.vals = calloc(d.X.rows, sizeof(double*));
+ d.X.vals = calloc(d.X.rows, sizeof(float*));
d.y = make_matrix(n, k);
for(i = 0; i < n; ++i){
@@ -106,8 +106,8 @@
data d;
d.shallow = 0;
matrix X = csv_to_matrix(filename);
- double *truth_1d = pop_column(&X, target);
- double **truth = one_hot_encode(truth_1d, X.rows, k);
+ float *truth_1d = pop_column(&X, target);
+ float **truth = one_hot_encode(truth_1d, X.rows, k);
matrix y;
y.rows = X.rows;
y.cols = k;
@@ -123,7 +123,7 @@
int i;
for(i = d.X.rows-1; i > 0; --i){
int index = rand()%i;
- double *swap = d.X.vals[index];
+ float *swap = d.X.vals[index];
d.X.vals[index] = d.X.vals[i];
d.X.vals[i] = swap;
@@ -156,10 +156,10 @@
train.X.cols = test.X.cols = d.X.cols;
train.y.cols = test.y.cols = d.y.cols;
- train.X.vals = calloc(train.X.rows, sizeof(double*));
- test.X.vals = calloc(test.X.rows, sizeof(double*));
- train.y.vals = calloc(train.y.rows, sizeof(double*));
- test.y.vals = calloc(test.y.rows, sizeof(double*));
+ train.X.vals = calloc(train.X.rows, sizeof(float*));
+ test.X.vals = calloc(test.X.rows, sizeof(float*));
+ train.y.vals = calloc(train.y.rows, sizeof(float*));
+ test.y.vals = calloc(test.y.rows, sizeof(float*));
for(i = 0; i < start; ++i){
train.X.vals[i] = d.X.vals[i];
--
Gitblit v1.10.0