HistradientBoostingClassifier с использованием категориальных переменных ⇐ Python

[Anonymous]

Я приземлился на HistradientBoostingClassifier, потому что, согласно документам, он обладает нативной поддержкой категориальных переменных без необходимости однопользового кодирования или кодирования метки. Я попытался запустить категориальные переменные через модель, но когда я подходил модель, я получил ошибку: < /p>
ValueError: could not convert string to float: 'A'
< /code>
Вот базовый пример кода того, что я пытаюсь сделать: < /p>
X = pd.DataFrame({
'feature_1': [1, 2, 3, 1, 2],
'feature_2': ['A', 'B', 'C', 'A', 'B']
})

y = pd.Series([0, 1, 0, 1, 0])

X['feature_2'] = X['feature_2'].astype('category')

categorical_features = [1]

model = HistGradientBoostingClassifier(categorical_features=categorical_features)

model.fit(X, y)
< /code>
У меня нет порядковых категориальных переменных, поэтому я не решаюсь выполнять кодирование метки. Я здесь чего -то не хватает?

Подробнее здесь: https://stackoverflow.com/questions/792 ... -variables