y_train_new = get_stilted_dataset(S_pqk, V_pqk, S_original, V_original)
y_test_new = get_stilted_dataset(S_pqk_test, V_pqk_test, S_test_original, V_test_original)

S_pqk_test, V_pqk_test = get_spectrum(
 tf.reshape(x_test_pqk, [-1, len(qubits) * 3]))
S_test_original, V_test_original = get_spectrum(
 tf.cast(x_test, tf.float32), gamma=0.005)
print('Eigenvectors of pqk kernel matrix for test:', V_pqk_test)
print('Eigenvectors of original kernel matrix for test:', V_test_original)