From bd92d3f2b1a339d392bb7ebb39d24a0c48b7eb7c Mon Sep 17 00:00:00 2001 From: Isaac De Vlugt Date: Thu, 30 Nov 2023 14:27:03 -0500 Subject: [PATCH] use tracker context --- .../tutorial_kernel_based_training.py | 17 +++++++++++------ 1 file changed, 11 insertions(+), 6 deletions(-) diff --git a/demonstrations/tutorial_kernel_based_training.py b/demonstrations/tutorial_kernel_based_training.py index 55d5b86d3e..4c93168d0f 100644 --- a/demonstrations/tutorial_kernel_based_training.py +++ b/demonstrations/tutorial_kernel_based_training.py @@ -288,10 +288,13 @@ def kernel_matrix(A, B): # Training the SVM optimizes internal parameters that basically # weigh kernel functions. # It is a breeze in scikit-learn, which is designed -# as a high-level machine learning library: +# as a high-level machine learning library (we'll put this in a +# ``qml.Tracker`` context so that we can +# track device executions): # -svm = SVC(kernel=kernel_matrix).fit(X_train, y_train) +with qml.Tracker(dev_kernel) as tracker: + svm = SVC(kernel=kernel_matrix).fit(X_train, y_train) ###################################################################### @@ -307,7 +310,7 @@ def kernel_matrix(A, B): # How many times was the quantum device evaluated? # -dev_kernel.num_executions +tracker.totals['executions'] ###################################################################### @@ -485,7 +488,9 @@ def quantum_model_predict(X_pred, trained_params, trained_bias): n_layers = 2 batch_size = 20 steps = 100 -trained_params, trained_bias, loss_history = quantum_model_train(n_layers, steps, batch_size) + +with qml.Tracker(dev_var) as tracker: + trained_params, trained_bias, loss_history = quantum_model_train(n_layers, steps, batch_size) pred_test = quantum_model_predict(X_test, trained_params, trained_bias) print("accuracy on test set:", accuracy_score(pred_test, y_test)) @@ -506,7 +511,7 @@ def quantum_model_predict(X_pred, trained_params, trained_bias): # How often was the device executed? # -dev_var.num_executions +tracker.totals['executions'] ###################################################################### @@ -688,4 +693,4 @@ def model_evals_nn(n_data, n_params, n_steps, split, batch_size): ############################################################################## # About the author # ---------------- -# .. include:: ../_static/authors/maria_schuld.txt \ No newline at end of file +# .. include:: ../_static/authors/maria_schuld.txt