NUP-2504: Add test to softmax overflow

ZQFORWARD · Apr 12, 2018 · 9475ec8 · 9475ec8
1 parent 669874c
commit 9475ec8
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/tests/unit/nupic/algorithms/sdr_classifier_test.py b/tests/unit/nupic/algorithms/sdr_classifier_test.py
@@ -24,6 +24,7 @@
 
 import cPickle as pickle
 import random
+import sys
 import tempfile
 import types
 import unittest2 as unittest
@@ -788,6 +789,16 @@ def testMultiStepPredictions(self):
     self.assertAlmostEqual(result2[0][1], 1.0, places=1)
 
 
+  def testSoftMaxOverflow(self):
+    """
+    Test if the softmax normalization overflows
+    """
+    c = SDRClassifier([1], 1.0, 0.1, 0)
+    weight = numpy.array([[sys.float_info.max_exp + 1]])
+    res = c.inferSingleStep([0], weight)
+    self.assertFalse(numpy.isnan(res), "SoftMax overflow")
+
+
   def _doWriteReadChecks(self, computeBeforeSerializing):
     c1 = SDRClassifier([0], 0.1, 0.1, 0)