Skip to content

Stojak139808/neural_network

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

14 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

neural_network

This repository has three implemenations of a simple neural network framwork, for teaching them using stochastic gradient descent.

First version uses vectors for most operation, which affected the speed. Second implementation (neural_network_v2) uses simple arrays for its operations and also the code is much clearer, becuase I first implemented needed matrix operations and then wrote SGD using these operations. Last version utilizes cuda, for making the learning process faster, this is mostly done by rewriting the Matrix.h header to run on GPU.

At the end, the GPU acceleration wasn't too impresive, because the network I've used is quite small and data transfer was too slow, even though it was limited to the absolute minimum. It did run faster for large networks, but they had poor accuracy for the case I've used (MNIST database and number recognition). The Acuracy peaked at about 96% which is still quite good considering the simple structure.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published