diff --git a/skqulacs/qkrr/qkrr.py b/skqulacs/qkrr/qkrr.py index d57a816..fda27c6 100644 --- a/skqulacs/qkrr/qkrr.py +++ b/skqulacs/qkrr/qkrr.py @@ -80,5 +80,5 @@ def predict(self, xs: NDArray[np.float_]) -> NDArray[np.float_]: x_qc = self.circuit.run(xs[i]) for j in range(len(self.data_states)): kar[i][j] = abs(inner_product(x_qc, self.data_states[j])) ** 2 - predicted: NDArray[np.float_] = self.krr.predict(kar) + predicted: NDArray[np.float_] = self.kernel_ridge_tuned.predict(kar) return predicted