This library is used for learning about neural networks. Reading material is from Neural Nets and Deep Learning and original source code is from this repository.
- Code for the confusion matrix can be found here in the
evalution
function.
- Code to visualize incorrect predictions can be found here in the
transform_digit
function. - For example, this number is a 9 but was predicted as a 7:
- Code to save and load pretrained models can be found here in the
save_network
andload_network
functions. - The
/src/pretrained_models
folder contains some pretrained models used during this project.
- The code to implement an ROC curve can be found here in the
roc
function.
- The experiments and their results can be found here.