From e4954e696e24ebbc70310c1b2f563ee4169f74ba Mon Sep 17 00:00:00 2001 From: Michelangelo Conserva Date: Mon, 29 May 2023 10:18:58 +0100 Subject: [PATCH] Increase default kernel ridge value to ensure positive-definite kernel matrix --- neural_testbed/generative/gp_regression_envlikelihood.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/neural_testbed/generative/gp_regression_envlikelihood.py b/neural_testbed/generative/gp_regression_envlikelihood.py index b56fbba..4d57001 100644 --- a/neural_testbed/generative/gp_regression_envlikelihood.py +++ b/neural_testbed/generative/gp_regression_envlikelihood.py @@ -36,7 +36,7 @@ def __init__(self, key: chex.PRNGKey, tau: int = 1, noise_std: float = 1, - kernel_ridge: float = 1e-6, + kernel_ridge: float = 1e-5, ntk: bool = False): # Checking the dimensionality of our data coming in.