From 54d744d3520add21ed633d005e0baa9d8302f323 Mon Sep 17 00:00:00 2001
From: Matthew Johnson <mattjj@csail.mit.edu>
Date: Thu, 4 Feb 2016 08:55:01 -0500
Subject: [PATCH] fix up pca, maybe, i should delete those functions
---
pyhsmm/util/plot.py | 7 +++----
1 file changed, 3 insertions(+), 4 deletions(-)
diff --git a/pyhsmm/util/plot.py b/pyhsmm/util/plot.py
index 98b7aa5..47c1248 100644
--- a/pyhsmm/util/plot.py
+++ b/pyhsmm/util/plot.py
@@ -1,8 +1,7 @@
from __future__ import division
import numpy as np
from matplotlib import pyplot as plt
-
-# TODO move pca to stats
+from stats import cov
def plot_gaussian_2D(mu, lmbda, color='b', centermarker=True,label='',alpha=1.,ax=None,artists=None):
'''
@@ -48,12 +47,12 @@ def pca_project_data(data,num_components=2):
def pca(data,num_components=2):
- U,s,Vh = np.linalg.svd(data - np.mean(data,axis=0))
+ U,s,Vh = np.linalg.svd(cov(data))
return Vh.T[:,:num_components]
def project_data(data,vecs):
- return np.dot(data,vecs.T)
+ return np.dot(data,vecs)
def project_ellipsoid(ellipsoid,vecs):
--
GitLab