Advertisement
Guest User

Untitled

a guest
Apr 9th, 2020
170
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.80 KB | None | 0 0
  1. def get_transformed_feature_names(column_transformer, columns):
  2.     features = []
  3.     for _, transformer, cols in column_transformer.transformers_:
  4.         if isinstance(transformer, OrdinalEncoder):
  5.             features.extend(cols)
  6.         elif isinstance(transformer, OneHotEncoder):
  7.             features.extend(transformer.get_feature_names(cols))
  8.         elif transformer == "passthrough":
  9.             features.extend(columns[cols])
  10.         else:
  11.             raise NotImplementedError(f"Transformer {transformer} is not implemented.")
  12.     return np.array(features)
  13.  
  14. select_pipeline.fit(titanic_train_X, titanic_train_y)
  15. feature_names = get_transformed_feature_names(select_pipeline['columntransformer'], titanic_train_X.columns)
  16.  
  17. print(feature_names[select_pipeline['selectkbest'].get_support()])
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement