Update TsetlinMachine.pyx

Author: olegranmo

TsetlinMachine.pyx

@@ -221,11 +221,10 @@ cdef class TsetlinMachine:
 
 				if self.clause_sign[j] >= 0:
 					# Type I Feedback				
-					self.feedback_to_clauses[j] += 1
-
-				elif self.clause_sign[j] < 0:
+					self.feedback_to_clauses[j] = 1
+				else:
 					# Type II Feedback
-					self.feedback_to_clauses[j] -= 1
+					self.feedback_to_clauses[j] = -1
 
 		elif y == 0:
 			for j in xrange(self.number_of_clauses):
@@ -234,11 +233,10 @@ cdef class TsetlinMachine:
 
 				if self.clause_sign[j] >= 0:
 					# Type II Feedback
-					self.feedback_to_clauses[j] -= 1
-
-				elif self.clause_sign[j] < 0:
+					self.feedback_to_clauses[j] = -1
+				else:
 					# Type I Feedback
-					self.feedback_to_clauses[j] += 1
+					self.feedback_to_clauses[j] = 1
 
 		for j in xrange(self.number_of_clauses):
 			if self.feedback_to_clauses[j] > 0: