one-hot 编码的生成与撤销

Posted K同学啊

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了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()))

以上是关于one-hot 编码的生成与撤销的主要内容，如果未能解决你的问题，请参考以下文章