making output of NN softmax instead os sigmoid
current training params