Update optuna_dashboard/preferential/samplers/gp.py

Co-authored-by: Naoto Mizuno <naotomizuno@preferred.jp>
This commit is contained in:
contramundum53
2023-09-08 14:10:47 +09:00
committed by GitHub
co-authored by Naoto Mizuno
parent 6039dd1f50
commit 348d92e012
+1 -1
View File
@@ -194,7 +194,7 @@ def _orthants_MVN_EP(
mu = mu - Sxy * ((db + mean1 * da) * dr)
cov = cov - (Sxy[:, None] * (da * dr)) @ Sxy[None, :]
log_zs[i] = logz
return (mu, cov, torch.sum(log_zs))
return mu, cov, torch.sum(log_zs)
_orthants_MVN_EP_jit = torch.jit.script(_orthants_MVN_EP)