def inverse_transform(self, X):
non_cat = pd.DataFrame(X[:, :len(self.non_cat_columns_)],
columns=self.non_cat_columns_)
cats = []
for col in self.cat_columns_:
slice_ = self.cat_blocks_[col]
categories = self.categories_map_[col]
ordered = self.ordered_map_[col]
codes = X[:, slice_].argmax(1)
series = pd.Series(pd.Categorical.from_codes(
codes, categories, ordered=ordered
), name=col)
cats.append(series)
df = pd.concat([non_cat] + cats, axis=1)[self.columns_]
return df
评论列表
文章目录