Model.py 文件源码

def select_mdl(self, mdl_type, param):
        """
        # define classifier and parameters
        :param mdl_type: specify which model to initialize
        :param param: a dict storing model parameters
        """
        if (mdl_type == 'xgb'):
            self.mdl = xgb.XGBClassifier(**param)
        elif (mdl_type == 'lr'):
            self.mdl = LogisticRegression(**param)
        elif (mdl_type == 'rf'):
            self.mdl = RandomForestClassifier(**param)