2 changed files with 1 additions and 36 deletions
+1 -1
View File
@@ -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, sizeof(buffer)/sizeof(MatrixType));
TEST_ASSERT_EQUAL_FLOAT_ARRAY(expectedResults, matrixToTest.buffer, matrixToTest.cols * matrixToTest.rows);
}
void setUp(void) {
-35
View File
@@ -5,45 +5,10 @@
#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)