Use sign function instead of heaviside step for activation function

gbroques · Apr 26, 2018 · 0c84f19 · 0c84f19
1 parent cafa4a4
commit 0c84f19
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/data.txt b/data.txt
@@ -1,6 +1,6 @@
 3 5
-1 1 1 0
-1 1 0 1
-0 0 1 1
-1 1 0 0
-1 0 0 1
+1 1 1 -1
+1 1 -1 1
+-1 -1 1 1
+1 1 -1 -1
+1 -1 -1 1
diff --git a/main.py b/main.py
@@ -13,7 +13,7 @@ def main():
     design_matrix = data[:, :num_features]
     target_values = data[:, -1]
 
-    perceptron = Perceptron(max_iter=100, learning_rate=0.2)
+    perceptron = Perceptron(max_iter=100, learning_rate=0.2, activation_function='sign')
     perceptron.fit(design_matrix, target_values)
 
     wait_for_test_example(perceptron, num_features)