From e3d84e3df5b047f6140a54a5aab0b62a5d619841 Mon Sep 17 00:00:00 2001
From: Matthew Johnson <mattjj@csail.mit.edu>
Date: Wed, 30 Mar 2016 14:17:14 -0400
Subject: [PATCH] replace function
---
pyhsmm/util/stats.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/pyhsmm/util/stats.py b/pyhsmm/util/stats.py
index 7fde828..7705d85 100644
--- a/pyhsmm/util/stats.py
+++ b/pyhsmm/util/stats.py
@@ -100,7 +100,7 @@ def whiten(datalist):
return general.treemap(apply_whitening, datalist)
def diag_whiten(datalist):
- mu, l = mean(datalist), np.sqrt(diag_of_cov(datalist))
+ mu, l = mean(datalist), np.sqrt(np.diag(cov(datalist)))
def apply_whitening(x):
return (x-mu)/l + mu
return general.treemap(apply_whitening, datalist)
--
GitLab