class LogisticExciteFunctionTest(unittest.TestCase): def setUp(self): self.fcn = LogisticExciteFunction(xMidpoint=5, maxValue=1, steepness=2) def testExcite(self): activation = numpy.array([0.0, 0.2, 0.4, 0.6, 0.8]) original = numpy.copy(activation) amount = 1.0 self.fcn.excite(activation, amount) for i in xrange(len(original)): self.assertGreater(activation[i], original[i]) def testDecayZeroActivation(self): activation = numpy.array([0.0, 0.0, 0.0, 0.0, 0.0]) expected = numpy.copy(activation) amount = 1.0 self.fcn.decay(activation, amount) npt.assert_allclose(activation, expected) def testExciteFullActivation(self): activation = numpy.array([1.0, 1.0, 1.0, 1.0, 1.0]) expected = numpy.copy(activation) amount = 10.0 self.fcn.excite(activation, amount) npt.assert_allclose(activation, expected)
def setUp(self): self.fcn = LogisticExciteFunction(xMidpoint=5, maxValue=1, steepness=2)