From 3be576b7b5c686ee8e1b53c467272f6097e17381 Mon Sep 17 00:00:00 2001 From: Zachary Frangella <60199061+zjf4@users.noreply.github.com> Date: Thu, 7 Nov 2024 21:20:06 -0800 Subject: [PATCH] Update svd.py Replace deprecated eigenvalues keyword in linalg.eigh by subset_by_index keyword. --- eigenpro3/utils/svd.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/eigenpro3/utils/svd.py b/eigenpro3/utils/svd.py index 3dd9620..d88d940 100644 --- a/eigenpro3/utils/svd.py +++ b/eigenpro3/utils/svd.py @@ -17,7 +17,7 @@ def nystrom_kernel_svd(samples, kernel_fn, top_q): kmat = kernel_fn(samples, samples).cpu().data.numpy() scaled_kmat = kmat / n_sample vals, vecs = linalg.eigh(scaled_kmat, - eigvals=(n_sample - top_q, n_sample - 1)) + subset_by_index=(n_sample - top_q, n_sample - 1)) eigvals = torch.from_numpy(vals).flip(0)[:top_q] eigvecs = torch.from_numpy(vecs).flip(1)[:, :top_q] beta = torch.from_numpy(kmat).diag().max()