decomposition.py 文件源码

def de_lda(X,y):
    """ lda """
    dim = X.shape[1]
    de  = min(2000,dim)
    clf = LDA(n_components = de)
    _,x_mini,_,y_mini = train_test_split(X,y,test_size = 0.33)
    clf.fit(x_mini,y_mini)
    def _func(X1,X2):
        return clf.transform(X1), clf.transform(X2)
    return _func

# def de_ps(X,y):
#     """ pearsonr method """
#     dim = X.shape[1]
#     de = min(2000,dim)
#     clf = SelectKBest(Pearsonr , k=de)
#     clf.fit(X,y)
#     def _func(X1,X2):
#         return clf.transform(X1),clf.transform(X2)
#     return _func