allowing nnpredict to pass nn back so we can get the activations from it