Codificando categorias
Você precisa ajudar seu algoritmo a entender que está lidando com categorias. Você vai codificar as categorias da variável salary, que você sabe ser ordinal com base nos valores observados:
- primeiro, você deve informar ao Python que a coluna
salaryé de fato categórica - depois, precisa especificar a ordem correta das categorias
- por fim, deve codificar cada categoria com um valor numérico que corresponda à sua posição nessa ordem
Este exercício faz parte do curso
HR Analytics: prevendo rotatividade de funcionários em Python
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
# Change the type of the "salary" column to categorical
data.salary = data.salary.____('category')