Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
fix decoder for multitarget encoder
  • Loading branch information
somefreestring committed Feb 12, 2020
commit a67c27983bc1a791c177f6b0f90c0ec844e38140
9 changes: 6 additions & 3 deletions pandas_ml_utils/model/features_and_labels/target_encoder.py
Original file line number Diff line number Diff line change
Expand Up @@ -71,7 +71,7 @@ def labels_source_columns(self) -> List[str]:

@property
def encoded_labels_columns(self) -> List[str]:
[l for enc in self.target_labels.values() for l in enc.encoded_labels_columns]
return [l for enc in self.target_labels.values() for l in enc.encoded_labels_columns]

def encode(self, df: pd.DataFrame, **kwargs) -> pd.DataFrame:
df_labels = pd.DataFrame({}, index=df.index)
Expand All @@ -81,8 +81,11 @@ def encode(self, df: pd.DataFrame, **kwargs) -> pd.DataFrame:
return df_labels

def decode(self, df: pd.DataFrame) -> pd.DataFrame:
# FIXME
pass
df_labels = pd.DataFrame({}, index=df.index)
for target, enc in self.target_labels.items():
df_labels = df_labels.join(enc.decode(df), how='inner', rsuffix=f'_{target}')

return df_labels

def __len__(self):
sum([len(enc) for enc in self.target_labels.values()])
Expand Down