metric support

This commit is contained in:
Arnar Flatberg 2007-03-14 16:33:16 +00:00
parent 48047f1395
commit 3bd21ab089

View File

@ -67,7 +67,7 @@ def pls_gen(a, b, n_blocks=None, center=False, index_out=False,axis=0, metric=No
yield acal, atrue, bcal, btrue
def pca_gen(a, n_sets=None, center=False, index_out=False, axis=0):
def pca_gen(a, n_sets=None, center=False, index_out=False, axis=0, metric=None):
"""Returns a generator of crossvalidation sample segments.
input:
@ -95,6 +95,8 @@ def pca_gen(a, n_sets=None, center=False, index_out=False, axis=0):
mn_a = acal.mean(0)[newaxis]
acal = acal - mn_a
atrue = atrue - mn_a
if metric!=None:
acal = dot(acal, metric)
if index_out:
yield acal, atrue, out
else: