Как вернуть одно горячее закодированное значение в метку? - PullRequest
0 голосов
/ 22 декабря 2018

Я работаю над простым набором данных, чтобы обнаружить рок или шахту с именами классов 'R' и 'M'.У меня есть одно горячее закодированное R в 1 и M в 0. Теперь я хочу раскрыть его.

Я пробовал много способов, но не смог найти подход для преобразования обратно в 1 в R и 0 в M

import numpy as np
import pandas as pd
import keras
from sklearn.preprocessing import LabelEncoder

df=pd.read_csv('D:\\Datasets\\node-fussy-examples-master\\node-fussy- 
examples-master\\sonar\\training.csv')
ds=df.values
x_train=df[df.columns[0:60]].values
y_train=df[df.columns[60]]

encoder = LabelEncoder()
encoder.fit(y_train)
encoded_Y = encoder.transform(y_train)

Я ожидаю, что 1 будет R и 0 будет M

1 Ответ

0 голосов
/ 22 декабря 2018

Вы можете использовать inverse_transform метод:

from sklearn import preprocessing
le = preprocessing.LabelEncoder()
le.fit([1, 2, 2, 6])
print(le.transform([1, 1, 2, 6]))
print(le.inverse_transform([0, 0, 1, 2]))

Если вам нужно сделать то же самое в Tensorflow, посмотрите на этот поток .

...