Train & predict for Neural Net