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

[Anonymous]

Я создаю классификатор случайного леса с помощью RandomForest и остановился на HistGradientBoostingClassifier, поскольку он имеет встроенную поддержку категориальных переменных без необходимости быстрого кодирования или кодирования меток. Я попытался запустить категориальные переменные через модель, но когда я подошел к модели, я получил ошибку:
"ValueError: не удалось преобразовать строку в число с плавающей запятой: 'A'
вот базовый пример кода того, что я пытаюсь сделать:

Код: Выделить всё

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)

У меня нет порядковых категориальных переменных, поэтому я не решаюсь использовать кодирование меток. Есть ли что-то, что мне здесь не хватает?

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