Compare commits
5
Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
fcdcef7e88 | ||
|
|
24c7ac65ce | ||
|
|
19a811515b | ||
|
|
734b72d346 | ||
|
|
f2b748a9c5 |
+1
-1
@@ -164,7 +164,7 @@ void test_setMatrixAtFailsOnIndicesOutOfRange(void)
|
||||
Matrix matrixToTest = {.rows=2, .cols=3, .buffer=buffer};
|
||||
|
||||
setMatrixAt(-1, matrixToTest, 2, 3);
|
||||
TEST_ASSERT_EQUAL_FLOAT_ARRAY(expectedResults, matrixToTest.buffer, matrixToTest.cols * matrixToTest.rows);
|
||||
TEST_ASSERT_EQUAL_FLOAT_ARRAY(expectedResults, matrixToTest.buffer, sizeof(buffer)/sizeof(MatrixType));
|
||||
}
|
||||
|
||||
void setUp(void) {
|
||||
|
||||
@@ -5,10 +5,45 @@
|
||||
#include "unity.h"
|
||||
#include "neuralNetwork.h"
|
||||
|
||||
#define FILE_HEADER_STRING "__info2_neural_network_file_format__"
|
||||
|
||||
static void prepareNeuralNetworkFile(const char *path, const NeuralNetwork nn)
|
||||
{
|
||||
// TODO
|
||||
|
||||
FILE *f = fopen(path, "wb");
|
||||
if (!f) return;
|
||||
|
||||
fwrite(FILE_HEADER_STRING, sizeof(char), strlen(FILE_HEADER_STRING), f);
|
||||
|
||||
int inputDim = nn.layers[0].weights.cols;
|
||||
int outputDim = nn.layers[0].weights.rows;
|
||||
|
||||
fwrite(&inputDim, sizeof(int), 1, f);
|
||||
fwrite(&outputDim, sizeof(int), 1, f);
|
||||
|
||||
for (int i = 0; i < nn.numberOfLayers; i++)
|
||||
{
|
||||
Matrix weights = nn.layers[i].weights;
|
||||
Matrix biases = nn.layers[i].biases;
|
||||
|
||||
int numWeightValues = weights.rows * weights.cols;
|
||||
int numBiasValues = biases.rows * biases.cols;
|
||||
|
||||
fwrite(weights.buffer, sizeof(MatrixType), numWeightValues, f);
|
||||
fwrite(biases.buffer, sizeof(MatrixType), numBiasValues, f);
|
||||
|
||||
if (i < nn.numberOfLayers - 1)
|
||||
{
|
||||
int nextOutputDim = nn.layers[i + 1].weights.rows;
|
||||
fwrite(&nextOutputDim, sizeof(int), 1, f);
|
||||
}
|
||||
}
|
||||
|
||||
int endMarker = 0;
|
||||
fwrite(&endMarker, sizeof(int), 1, f);
|
||||
|
||||
fclose(f);
|
||||
}
|
||||
|
||||
void test_loadModelReturnsCorrectNumberOfLayers(void)
|
||||
|
||||
Reference in New Issue
Block a user