batch train

This commit is contained in:
Thomas 2023-09-23 21:00:23 +02:00
parent 66ed7afb9f
commit 45f39130c1
3 changed files with 16 additions and 16 deletions

9
main.c
View file

@ -7,17 +7,12 @@ int main() {
Image** images = import_images("../data/train-images.idx3-ubyte", "../data/train-labels.idx1-ubyte", NULL, 60000);
// img_visualize(images[0]);
Neural_Network* nn = new_network(28*28, 100, 5, 10, 0.01);
Neural_Network* nn = new_network(28*28, 32, 3, 10, 0.01);
randomize_network(nn, 10);
// save_network(nn);
// Neural_Network* nn = load_network("../networks/test1.txt");
for (int i = 0; i < 60000; ++i) {
train_network(nn, images[i], images[i]->label);
}
// train_network(nn, images[0], images[0]->label);
// train_network(nn, images[0], images[0]->label);
batch_train(nn, images, 20000, 16);
printf("%lf\n", measure_network_accuracy(nn, images, 10000));