Interesting recurrent configuration

cair · May 9, 2024 · c75110e · c75110e
1 parent cf5db92
commit c75110e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/classification/RecurrentInterpretabilityDemo.py b/examples/classification/RecurrentInterpretabilityDemo.py
@@ -34,7 +34,7 @@ def main(args):
     Y_test = np.logical_xor(X_test[:, 0], X_test[:, 1]).astype(dtype=np.uint32)
     X_test = X_test.reshape(-1, 1, args.number_of_features)
 
-    tm = TMClassifier(args.number_of_clauses, args.T, args.s, number_of_state_bits_ta=6, patch_dim=(1, 1), weighted_clauses=False, platform=args.platform, boost_true_positive_feedback=True, recurrent=True, incremental=False)
+    tm = TMClassifier(args.number_of_clauses, args.T, args.s, number_of_state_bits_ta=6, patch_dim=(1, 1), weighted_clauses=True, platform=args.platform, boost_true_positive_feedback=True, recurrent=True, incremental=False)
 
     for i in range(args.epochs):
         tm.fit(X_train, Y_train)
@@ -143,7 +143,7 @@ def default_args(**kwargs):
     parser.add_argument("--epochs", default=50, type=int)
     parser.add_argument("--number-of-clauses", default=10, type=int)
     parser.add_argument("--platform", default='CPU', type=str)
-    parser.add_argument("--T", default=5, type=int)
+    parser.add_argument("--T", default=5*10, type=int)
     parser.add_argument("--s", default=2.5, type=float)
     parser.add_argument("--number-of-features", default=6, type=int)
     parser.add_argument("--noise", default=0.1, type=float, help="Noisy XOR")