diff options
| author | 3gg <3gg@shellblade.net> | 2022-05-12 20:31:56 -0700 |
|---|---|---|
| committer | 3gg <3gg@shellblade.net> | 2022-05-12 20:31:56 -0700 |
| commit | e858458e934a9e2fca953be43120497771292304 (patch) | |
| tree | 56252dd04ad07e2bf4d1ab42c3ca0a0d7d5dbcd9 /src/bin/mnist | |
| parent | 411f66a2540fa17c736116d865e0ceb0cfe5623b (diff) | |
Fix typo in OneHotDecode.
Diffstat (limited to 'src/bin/mnist')
| -rw-r--r-- | src/bin/mnist/src/main.c | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/bin/mnist/src/main.c b/src/bin/mnist/src/main.c index 4d268ac..9aa3ce5 100644 --- a/src/bin/mnist/src/main.c +++ b/src/bin/mnist/src/main.c | |||
| @@ -252,8 +252,8 @@ static void OneHotEncode(const uint8_t* labels_bytes, int num_labels, nnMatrix* | |||
| 252 | 252 | ||
| 253 | static int OneHotDecode(const nnMatrix* label_matrix) { | 253 | static int OneHotDecode(const nnMatrix* label_matrix) { |
| 254 | assert(label_matrix); | 254 | assert(label_matrix); |
| 255 | assert(label_matrix->cols == 1); | 255 | assert(label_matrix->cols == 10); |
| 256 | assert(label_matrix->rows == 10); | 256 | assert(label_matrix->rows == 1); |
| 257 | 257 | ||
| 258 | R max_value = 0; | 258 | R max_value = 0; |
| 259 | int pos_max = 0; | 259 | int pos_max = 0; |