Fix softmax overflow

numenta · Mar 13, 2018 · 08daff4 · 08daff4
1 parent 9bb7705
commit 08daff4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/nupic/algorithms/sdr_classifier.py b/src/nupic/algorithms/sdr_classifier.py
@@ -374,6 +374,7 @@ def inferSingleStep(self, patternNZ, weightMatrix):
     outputActivation = weightMatrix[patternNZ].sum(axis=0)
 
     # softmax normalization
+    outputActivation = outputActivation - numpy.max(outputActivation)
     expOutputActivation = numpy.exp(outputActivation)
     predictDist = expOutputActivation / numpy.sum(expOutputActivation)
     return predictDist