mirror of
https://github.com/wassname/optuna-dashboard.git
synced 2026-09-25 13:50:21 +08:00
Update optuna_dashboard/preferential/samplers/gp.py
Co-authored-by: Naoto Mizuno <naotomizuno@preferred.jp>
This commit is contained in:
co-authored by
Naoto Mizuno
parent
6039dd1f50
commit
348d92e012
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user