metric support
This commit is contained in:
parent
48047f1395
commit
3bd21ab089
@ -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
|
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.
|
"""Returns a generator of crossvalidation sample segments.
|
||||||
|
|
||||||
input:
|
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]
|
mn_a = acal.mean(0)[newaxis]
|
||||||
acal = acal - mn_a
|
acal = acal - mn_a
|
||||||
atrue = atrue - mn_a
|
atrue = atrue - mn_a
|
||||||
|
if metric!=None:
|
||||||
|
acal = dot(acal, metric)
|
||||||
if index_out:
|
if index_out:
|
||||||
yield acal, atrue, out
|
yield acal, atrue, out
|
||||||
else:
|
else:
|
||||||
|
Reference in New Issue
Block a user