Update

cair · Feb 9, 2024 · ed4d1b3 · ed4d1b3
1 parent e355b28
commit ed4d1b3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/autoencoder/DimensionalityReductionDemo.py b/examples/autoencoder/DimensionalityReductionDemo.py
@@ -27,7 +27,7 @@
 X_train = np.random.randint(0, 2, size=(5000, number_of_features), dtype=np.uint32)
 X_train[:,0] = np.where(np.random.rand(5000) <= noise, 1- X_train[:,1],  X_train[:,1])
 
-tm = TMAutoEncoder(clauses, T, s, output_active, max_included_literals=3, accumulation=accumulation, feature_negation=True, platform='CPU', output_balancing=True)
+tm = TMAutoEncoder(clauses, T, s, output_active, max_included_literals=3, accumulation=accumulation, feature_negation=True, platform='CPU', output_balancing=0.5)
 
 print("\nAccuracy Over 40 Epochs:")
 for e in range(40):