Merge branch 'main' of github.com:quicophy/mdopt

examples/decoding/classical.py

@@ -53,9 +53,11 @@
 parser = argparse.ArgumentParser(
     description="Launch calculations on Compute Canada clusters."
 )
-parser.add_argument("system_size", type=int, help="System size as the number of bits.")
 parser.add_argument(
-    "bond_dim", type=int, help="Maximum bond dimension to keep during contraction."
+    "--system_size", type=int, help="System size as the number of bits."
+)
+parser.add_argument(
+    "--bond_dim", type=int, help="Maximum bond dimension to keep during contraction."
 )
 args = parser.parse_args()
 
@@ -125,7 +127,7 @@
                 chi_max=CHI_MAX_CONTRACTOR,
                 renormalise=True,
                 result_to_explicit=False,
-                strategy="Naive",
+                strategy="Optimized",
                 silent=True,
             )
             logging.info("Decoding the perturbed codeword state using DMRG.")

examples/decoding/decoding.py

@@ -714,6 +714,8 @@ def apply_constraints(
     entropies = []
     bond_dims = []
 
+    # Using matrix front minimization technique to optimize the order
+    # in which to apply the checks.
     if strategy == "Optimized":
         mpo_location_matrix = np.zeros((len(strings), mps.num_sites))
         for row_idx, sublist in enumerate(strings):
@@ -724,6 +726,7 @@ def apply_constraints(
         optimized_order = msro(mpo_location_matrix)
         strings = [strings[index] for index in optimized_order]
 
+    # Do not optimize the order in which to apply the checks.
     if strategy == "Naive":
         pass