Невозможно передать категориальную переменную в Random Forest

Question

У меня есть 10 категориальных переменных и 3 числовые переменные. Я непосредственно передал их в Случайный Лес после разделения, но это показывает,

ValueError: could not convert string to float: 'No internet service'

Я прочитал, что Random Forest может обрабатывать как числовые, так и категориальные переменные. Почему это показывает эту ошибку?

Answer 1

Это зависит от библиотеки, которую вы используете для реализации RandomForest. Например, R может обрабатывать категориальные переменные с помощью RandomForest. Однако из моего опыта Sci-kit в Python требует преобразования в фиктивные переменные. Ошибка говорит мне, что вы используете Python, поэтому вы, скорее всего, будете использовать sklearn, который требует преобразования в фиктивные переменные. Вы можете конвертировать ваш столбец с помощью одной строки кода.

pd.get_dummies(df['column_name'])