PIP

Author: DuckTigger

Diff for: qutrits/qutrit_ops.py

@@ -4,6 +4,7 @@
 from cirq.ops import gate_features, eigen_gate
 from sympy.physics.quantum import TensorProduct
 
+
 class QutritPlusGate(eigen_gate.EigenGate, gate_features.SingleQubitGate):
     def _qid_shape_(self):
         return (3,)

Diff for: run.py

@@ -5,6 +5,7 @@
 
 from encode_state import EncodeState
 from leakage import LeakageModels
+from qutrit_model import QutritModel
 from input_circuits import InputCircuits
 from loss import DiscriminationLoss
 from noise.noise_model import TwoQubitNoiseModel, two_qubit_depolarize
@@ -16,14 +17,16 @@ def main():
     train_circuits, train_labels, test_circuits, test_labels = circuits.create_discrimination_circuits(mu_a=0.9)
     encoder = EncodeState(n)
     leakage = LeakageModels(2, 2, False, 0.3)
+    qutrits = QutritModel(2, 0.1)
     noise_model = TwoQubitNoiseModel(cirq.depolarize(0.01), two_qubit_depolarize(0.01))
     noisy_sim = cirq.DensityMatrixSimulator(noise=noise_model)
     # pqc_model = encoder.encode_state_PQC()
     # discrimination_model = encoder.discrimination_model()
     # controlled_model = encoder.discrimination_model(True)
     # noisy_discrimination = encoder.discrimination_model(backend=noisy_sim)
-    leakage_model = leakage.leaky_model()
-    model = leakage_model
+    # leakage_model = leakage.leaky_model()
+    qutrit_model = qutrits.qutrit_model()
+    model = qutrit_model
 
     loss = DiscriminationLoss(0.5, 0.5)
     loss_fn = loss.discrimination_loss