Source code for tramp.beliefs.sparse

import numpy as np
from scipy.special import expit
from . import normal

[docs]def A(a, b, eta): return np.logaddexp(eta, normal.A(a, b))
[docs]def p(a, b, eta): xi = normal.A(a, b) - eta s = expit(xi) return s
[docs]def r(a, b, eta): s = p(a, b, eta) return s * (b / a)
[docs]def v(a, b, eta): s = p(a, b, eta) return s / a + s * (1-s) * (b / a)**2
[docs]def tau(a, b, eta): s = p(a, b, eta) return s / a + s * (b / a)**2