compute for NN is ok

compute for NN is ok , predict when given weights of the different
layers, Next is the training of the NN to get these weights.