one-hot 编码的生成与撤销

生成one-hot 编码

from sklearn import preprocessing
import  numpy as np

enc = preprocessing.OneHotEncoder(categories='auto')
# 训练onehot编码，指定标签
enc.fit([[1],[2],[3]])

# 将标签转换成 onehot编码
result =enc.transform([[1],[3],[2]])
print(result.toarray())

撤销one-hot 编码

print(enc.inverse_transform(result.toarray()))

---