Advertisement
Guest User

Untitled

a guest
Jul 23rd, 2019
83
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.64 KB | None | 0 0
  1. from sklearn.preprocessing import OrdinalEncoder
  2. enc = OrdinalEncoder()
  3. X = [['Male', 1], ['Female', 3], ['Female', 2]]
  4. enc.fit(X)
  5.  
  6. print(enc.categories_)
  7. #Output: [array(['Female', 'Male'], dtype=object), array([1, 2, 3], dtype=object)]
  8.  
  9. encoding = enc.categories_
  10. encoding_sex = dict(zip((encoding[0]), range(len(encoding[0]))))
  11. print(encoding_sex)
  12. # Output: {'Female': 0, 'Male': 1}
  13.  
  14. encoding = enc.categories_
  15. encoding_feature = lambda x: dict(zip(x, range(len(x))))
  16. encoding_full = [encoding_feature(feature_elem) for feature_elem in encoding]
  17. print(encoding_full)
  18. # Output: [{'Female': 0, 'Male': 1}, {1: 0, 2: 1, 3: 2}]
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement