Формирование custom_transformer для фрейма данных - PullRequest
Новая
       26

Формирование custom_transformer для фрейма данных

0 голосов
/ 25 мая 2020

Я хочу сформировать настраиваемый преобразователь, который добавляет новый столбец ',' Evaluation 'в фрейм данных.

Например, если' оценка 'чрезвычайно рискованна,' Evaluation 'будет равно 1, иначе 0.

Но когда я запускаю приведенный ниже код, я получаю' ValueError: истинное значение серии неоднозначно. Используйте a.empty, a.bool (), a.item (), a.any () или a.all () '

Знаете ли вы, какие могут быть возможные причины?

класс Risk_Transformer (TransformerMixin, BaseEstimator): 

 def fit(self, X_final, y=None):
   self.age = X_final['Age_condition']
   self.BMI = X_final["BMI_condition"] 
   self.weight =X_final['Weight_condition']
   self.Risk=X_final['Risk Evaluation']

def transform(self, X, y=None):

    if self.Risk =='extremely_risky':
      X.loc[:,'Evaluation'] =1
    else:
      X.loc[:,'Evaluation'] =0
      del X['Risk Evaluation']

    return X
    pass

risk_transformer = Risk_Transformer () risk_transformer.fit (X_final)

risk_transformer.transform (X_final) X_final.head (5)

...