Skip to content

Commit c0742ec

Browse files
committed
Merge pull request hildensia#7 from pedronariyoshi/patch-1
Corrected scale and beta factor calculation
2 parents 63771d2 + c1d2c87 commit c0742ec

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

bayesian_changepoint_detection/online_changepoint_detection.py

+3-5
Original file line numberDiff line numberDiff line change
@@ -51,19 +51,17 @@ def pdf(self, data):
5151
return stats.t.pdf(x=data,
5252
df=2*self.alpha,
5353
loc=self.mu,
54-
scale=self.beta * (self.kappa+1) / (self.alpha *
55-
self.kappa))
54+
scale=np.sqrt(self.beta * (self.kappa+1) / (self.alpha *
55+
self.kappa)))
5656

5757
def update_theta(self, data):
5858
muT0 = np.concatenate((self.mu0, (self.kappa * self.mu + data) / (self.kappa + 1)))
5959
kappaT0 = np.concatenate((self.kappa0, self.kappa + 1.))
6060
alphaT0 = np.concatenate((self.alpha0, self.alpha + 0.5))
61-
betaT0 = np.concatenate((self.beta0, self.kappa + (self.kappa * (data -
61+
betaT0 = np.concatenate((self.beta0, self.beta + (self.kappa * (data -
6262
self.mu)**2) / (2. * (self.kappa + 1.))))
6363

6464
self.mu = muT0
6565
self.kappa = kappaT0
6666
self.alpha = alphaT0
6767
self.beta = betaT0
68-
69-

0 commit comments

Comments
 (0)