Advertisement
Guest User

Untitled

a guest
Nov 22nd, 2017
85
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.58 KB | None | 0 0
  1. from copy import deepcopy
  2.  
  3. def dummy_encode_categorical_columns(df, categorical_columns):
  4.     result_df = deepcopy(df)
  5.    #print result_df
  6.     for column in categorical_columns:
  7.         result_df = pd.concat([result_df, pd.get_dummies(result_df[column],
  8.             prefix=column, prefix_sep=': ')], axis=1)
  9.         del result_df[column]
  10.     return result_df
  11.  
  12. categorical_columns = set(df_train.columns[df_train.dtypes == 'object'])
  13. #print 'before',df_train
  14. df_no_cat = dummy_encode_categorical_columns(df_train, categorical_columns)
  15. X_train=df_no_cat.values.astype(np.float32)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement