jadrian.py 文件源码

def __init__( self, pdr, E_QC = "E_QC", Em = "Em", type_name = "Type", type_l = [1,2,3,4], 
                    disp = False, graph = False):

        # This parameter will be used in the run() function. 
        self.type_l = type_l
        self.disp = disp
        self.graph = graph

        self.xMa = {}
        self.yVa = {}
        # self.kfa = {}
        for type_id in type_l:
            pdr_new = pdr[ pdr[ type_name] == type_id]
            self.xMa[type_id] = np.mat( pdr_new[ E_QC].values).T
            self.yVa[type_id] = np.mat( pdr_new[ Em].values).T
            # kfa[type_id] = cross_validation.KFold( np.shape(yVa[type_id])[0], n_folds=5, shuffle=True)