Update

cair · Feb 23, 2022 · 923b721 · 923b721
1 parent 48a9893
commit 923b721
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/examples/MNISTDemoTypeIIIFeedback.py b/examples/MNISTDemoTypeIIIFeedback.py
@@ -5,16 +5,17 @@
 
 from tmu.tsetlin_machine import TMClassifier
 
-number_of_clauses = 2000
-T = number_of_clauses*2.5
+clauses = 2000
+T = 5000
 s = 10.0
+number_of_state_bits_ta = 8
 
 (X_train, Y_train), (X_test, Y_test) = mnist.load_data()
 
 X_train = np.where(X_train.reshape((X_train.shape[0], 28*28)) > 75, 1, 0) 
 X_test = np.where(X_test.reshape((X_test.shape[0], 28*28)) > 75, 1, 0) 
 
-tm = TMClassifier(number_of_clauses, T, s, type_iii_feedback=True, number_of_state_bits_ta=8, number_of_state_bits_ind=8, platform='CPU', weighted_clauses=True)
+tm = TMClassifier(clauses, T, s, type_iii_feedback=True, number_of_state_bits_ta=number_of_state_bits_ta, number_of_state_bits_ind=8, platform='CPU', weighted_clauses=True)
 
 f = open("mnist_type_iii_feedback_%.1f_%d_%d_%d.txt" % (s, clauses, T, number_of_state_bits_ta), "w+")
 print("\nAccuracy over 250 epochs:\n")