From 2ad3fa9c7081a2cd892745e5eef292944aa610b0 Mon Sep 17 00:00:00 2001 From: Maxim Ziatdinov <34245227+ziatdinovmax@users.noreply.github.com> Date: Wed, 7 Feb 2024 19:29:49 +0000 Subject: [PATCH] Update docstrings --- gpax/models/sparse_gp.py | 8 ++++++-- gpax/models/vigp.py | 8 ++++++-- 2 files changed, 12 insertions(+), 4 deletions(-) diff --git a/gpax/models/sparse_gp.py b/gpax/models/sparse_gp.py index c336180..37f0210 100644 --- a/gpax/models/sparse_gp.py +++ b/gpax/models/sparse_gp.py @@ -37,8 +37,12 @@ class viSparseGP(viGP): Optional custom priors over kernel hyperparameters; uses LogNormal(0,1) by default mean_fn_prior: Optional priors over mean function parameters - noise_prior: - Optional custom prior for the observation noise variance; uses LogNormal(0,1) by default. + noise_prior_dist: + Optional custom prior distribution over the observational noise variance. + Defaults to LogNormal(0,1). + lengthscale_prior_dist: + Optional custom prior distribution over kernel lengthscale. + Defaults to LogNormal(0, 1). guide: Auto-guide option, use 'delta' (default) or 'normal' """ diff --git a/gpax/models/vigp.py b/gpax/models/vigp.py index 11f8de3..e734441 100644 --- a/gpax/models/vigp.py +++ b/gpax/models/vigp.py @@ -35,8 +35,12 @@ class viGP(ExactGP): Optional custom priors over kernel hyperparameters; uses LogNormal(0,1) by default mean_fn_prior: Optional priors over mean function parameters - noise_prior: - Optional custom prior for the observation noise variance; uses LogNormal(0,1) by default. + noise_prior_dist: + Optional custom prior distribution over the observational noise variance. + Defaults to LogNormal(0,1). + lengthscale_prior_dist: + Optional custom prior distribution over kernel lengthscale. + Defaults to LogNormal(0, 1). guide: Auto-guide option, use 'delta' (default) or 'normal'