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.