signature_genes.py 文件源码

def get_signature_genes(X,n,lda=10):
    W = np.zeros((X.shape[0],X.shape[0]))
    # coarse search from the bottom
    while (abs(W).sum(1) > 0).sum() < n:
        lda /= 10.
        model = MultiTaskLasso(alpha=lda,max_iter=100,tol=.001,selection='random',warm_start=True)
        model.fit(X.T,X.T)
        W = model.coef_.T
        #print len(np.nonzero(abs(W).sum(1))[0]),model.score(X.T,X.T)
    # fine search from the top
    while (abs(W).sum(1) > 0).sum() > n*1.2:
        lda *= 2.
        model.set_params(alpha=lda)
        model.fit(X.T,X.T)
        W = model.coef_.T
        #print len(np.nonzero(abs(W).sum(1))[0]),model.score(X.T,X.T)
    # finer search
    while (abs(W).sum(1) > 0).sum() > n:
        lda *= 1.1
        model.set_params(alpha=lda)
        model.fit(X.T,X.T)
        W = model.coef_.T
        #print len(np.nonzero(abs(W).sum(1))[0]),model.score(X.T,X.T)
    return np.nonzero(abs(W).sum(1))[0]