OneHotEncoder не ожидал аргумента ключевого слова 'categoryorical_features'

Question

Я пытаюсь запустить программу распознавания лиц . В разделе read me было написано, что я должен запустить train_softmax.py, но в коде есть этот раздел:

# Encode the labels
le = LabelEncoder()
labels = le.fit_transform(data["names"])
num_classes = len(np.unique(labels))
labels = labels.reshape(-1, 1)
one_hot_encoder = OneHotEncoder(categorical_features = [0])
labels = one_hot_encoder.fit_transform(labels).toarray()

, который вызывает ошибку:

__init__() got an unexpected keyword argument 'categorical_features'

, но изменение 'categoryorical_features' на 'category' или использование ColumnTransformer вызывает другие ошибки. что мне делать? t

Answer 1

замена всего этого блока

#le = LabelEncoder()
#labels = le.fit_transform(data["names"])
#num_classes = len(np.unique(labels))
#labels = labels.reshape(-1, 1)
#one_hot_encoder = OneHotEncoder(categorical_features = [0])
#labels = one_hot_encoder.fit_transform(labels).toarray()

ниже приведет к тому же результату, что и LabelEncoder and OneHotEncoder объединить

from sklearn.compose import ColumnTransformer

num_classes = len(np.unique(data["names"])) 

ct = ColumnTransformer([('myCT', OneHotEncoder(), [0])])
labels = ct.fit_transform(df[['names']])