learner_nn function

Function of the neural network model (nn)