#6 Add the callback function

ksk0629 · Oct 18, 2024 · e10c08b · e10c08b
1 parent b61267b
commit e10c08b
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/scripts/run_example_estimator_qnn.py b/scripts/run_example_estimator_qnn.py
@@ -4,7 +4,7 @@
 from sklearn.model_selection import train_test_split
 
 from src.qnn_builder import QNNBuilder
-from src.utils import fix_seed, generate_line_dataset
+from src.utils import fix_seed, generate_line_dataset, callback_print
 
 if __name__ == "__main__":
     # Fix the random seed.
@@ -26,6 +26,7 @@
     classifier = NeuralNetworkClassifier(
         example_estimator_qnn,
         optimizer=qiskit_algorithms.optimizers.COBYLA(maxiter=200),
+        callback=callback_print,
     )
     # Fit the model.
     x = np.asarray(train_images)

diff --git a/scripts/run_example_exact_aer_estimator_qnn.py b/scripts/run_example_exact_aer_estimator_qnn.py
@@ -4,7 +4,7 @@
 from sklearn.model_selection import train_test_split
 
 from src.qnn_builder import QNNBuilder
-from src.utils import fix_seed, generate_line_dataset
+from src.utils import fix_seed, generate_line_dataset, callback_print
 
 if __name__ == "__main__":
     # Fix the random seed.
@@ -26,6 +26,7 @@
     classifier = NeuralNetworkClassifier(
         example_estimator_qnn,
         optimizer=qiskit_algorithms.optimizers.COBYLA(maxiter=200),
+        callback=callback_print,
     )
     # Fit the model.
     x = np.asarray(train_images)

diff --git a/scripts/run_example_noisy_aer_estimator_qnn.py b/scripts/run_example_noisy_aer_estimator_qnn.py
@@ -4,7 +4,7 @@
 from sklearn.model_selection import train_test_split
 
 from src.qnn_builder import QNNBuilder
-from src.utils import fix_seed, generate_line_dataset
+from src.utils import fix_seed, generate_line_dataset, callback_print
 
 if __name__ == "__main__":
     # Fix the random seed.
@@ -26,6 +26,7 @@
     classifier = NeuralNetworkClassifier(
         example_estimator_qnn,
         optimizer=qiskit_algorithms.optimizers.COBYLA(maxiter=200),
+        callback=callback_print,
     )
     # Fit the model.
     x = np.asarray(train_images)