Corrected step function

AbstractGeek · Feb 9, 2017 · 1eb34a3 · 1eb34a3
1 parent c1c6617
commit 1eb34a3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/libraries/ANN.py b/libraries/ANN.py
@@ -297,10 +297,10 @@ def sigmoid(self, x):
     def sigmoid_diff(self, output):
         ''' This is derivative of the sigmoid activation function. '''
         return output*(1-output)
-    
+
     def step(self, x, theta=0):
         ''' This is sigmoid activation function. '''
-        return 1 if x>theta else -1
+        return 1 if x >= theta else -1
 
     # Hyperbolic tan activation #
     def tanh(self, x):

diff --git a/packages/ANN/__init__.py b/packages/ANN/__init__.py
@@ -297,10 +297,10 @@ def sigmoid(self, x):
     def sigmoid_diff(self, output):
         ''' This is derivative of the sigmoid activation function. '''
         return output*(1-output)
-    
+
     def step(self, x, theta=0):
         ''' This is sigmoid activation function. '''
-        return 1 if x>theta else -1
+        return 1 if x >= theta else -1
 
     # Hyperbolic tan activation #
     def tanh(self, x):