0

sklearn の OneHotEncoder を使用していますが、データの変換を解除したいと考えています。それを行う方法はありますか?

>>> from sklearn.preprocessing import OneHotEncoder
>>> enc = OneHotEncoder()
>>> enc.fit([[0, 0, 3], [1, 1, 0], [0, 2, 1], [1, 0, 2]])  
>>> enc.n_values_
array([2, 3, 4])
>>> enc.feature_indices_
array([0, 2, 5, 9])
>>> enc.transform([[0, 1, 1]]).toarray()
array([[ 1.,  0.,  0.,  1.,  0.,  0.,  1.,  0.,  0.]])

しかし、私は次のことができるようにしたい:

>>> enc.untransform(array([[ 1.,  0.,  0.,  1.,  0.,  0.,  1.,  0.,  0.]]))
[[0, 1, 1]]

どうすればこれを行うことができますか?

コンテキストとして、ワンホット エンコーディング スペースを学習するニューラル ネットワークを構築しました。今度は nn を使用して、元のデータ形式である必要がある実際の予測を行いたいと考えています。

4

1 に答える 1