Кодирование категорий
Нужно помочь алгоритму понять, что он работает с категориальными данными. Вы закодируете категории переменной salary, которая, судя по наблюдаемым значениям, является порядковой:
- сначала нужно указать Python, что столбец
salaryявляется категориальным; - затем задать правильный порядок категорий;
- наконец, присвоить каждой категории числовое значение, соответствующее её позиции в этом порядке.
Это упражнение является частью курса
HR-аналитика: прогнозирование текучести кадров на Python
Интерактивное практическое упражнение
Попробуйте выполнить это упражнение, дополнив этот пример кода.
# Change the type of the "salary" column to categorical
data.salary = data.salary.____('category')