Пример #1
0
 def create_features(self):
     folds = pd.read_feather('../folds/03_stkfold.feather')
     train['position_education'] = train['position'].astype(str) + '_' + train['education'].astype(str)
     test['position_education'] = test['position'].astype(str) + '_' + test['education'].astype(str)
     tr_feat, te_feat = target_encoding(train, test, 'salary', 'position_education', folds)
     self.train[self.__class__.__name__] = tr_feat
     self.test[self.__class__.__name__] = te_feat
Пример #2
0
 def create_features(self):
     tr_feat, te_feat = target_encoding(train, test, 'meter_reading',
                                        'primary_use', folds2)
     self.train[self.__class__.__name__] = np.log1p(tr_feat)
     self.test[self.__class__.__name__] = np.log1p(te_feat)
Пример #3
0
 def create_features(self):
     folds = pd.read_feather('../folds/03_stkfold.feather')
     tr_feat, te_feat = target_encoding(train, test, 'salary', 'sex', folds)
     self.train[self.__class__.__name__] = tr_feat
     self.test[self.__class__.__name__] = te_feat
Пример #4
0
 def create_features(self):
     tr_feat, te_feat = target_encoding(train, test, 'meter_reading',
                                        'site_id', folds1)
     self.train[self.__class__.__name__] = tr_feat
     self.test[self.__class__.__name__] = te_feat