softmax used
training set generation testing
bug fixed