batch train
This commit is contained in:
parent
66ed7afb9f
commit
45f39130c1
3 changed files with 16 additions and 16 deletions
9
main.c
9
main.c
|
|
@ -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));
|
||||
|
||||
|
|
|
|||
Reference in a new issue