FeatureEngineering.py 文件源码

def __LaunchTask(self, task, PredCols):
        """"""
        print('\n---- Begin to deal with %s' % (task))
        start = time.time()

        if(task == 'MissingValue'):
            self.TrainData, self.TestData = self._missing.impute((self.TrainData, self.TestData))
        elif(task == 'NewFeature'):
            self.TrainData, self.TestData = self._newfeat.create((self.TrainData, self.TestData), PredCols)
        elif(task == 'FeatureEncoding'):
            with open('%s/featmap.pkl' % self._InputDir, 'rb') as i_file:
                d_feat = pickle.load(i_file)
            i_file.close()
            self.TrainData, self.TestData = self._encoding.ordinal((self.TrainData, self.TestData), d_feat)
        elif(task == 'FeatureSelection'):
            self.TrainData, self.TestData = self._select.select((self.TrainData, self.TestData))

        end = time.time()
        print('---- Task %s done, time consumed %ds' % (task, (end - start)))