arpachat's picture
Changed probability calculation by removing the extra softmax function
4cf4218