Aprimorando o gráfico
Para deixar o gráfico mais legível, precisamos atingir dois objetivos:
- Reordenar as barras em ordem crescente.
- Adicionar rótulos ao gráfico que correspondam aos nomes das features.
Para isso, vamos aproveitar a indexação do NumPy. O método .argsort() ordena um array e retorna os índices. Vamos usar esses índices para alcançar ambos os objetivos.
Este exercicio faz parte do curso
Análise de Marketing: Prevendo Churn de Clientes em Python
Instruções do exercicio
- Calcule os índices ordenados de
importancesusandonp.argsort()emimportances. - Crie os rótulos ordenados acessando as colunas de
Xe indexando porsorted_index. - Crie o gráfico indexando
importancesusandosorted_indexe especificando o argumento nomeadotick_label=labels.
exercicio interativo prático
Tente este exercicio completando este código de exemplo.
# Sort importances
sorted_index = ____(____)
# Create labels
labels = X.columns[____]
# Clear current plot
plt.clf()
# Create plot
plt.barh(range(X.shape[1]), importances[____], tick_label=____)
plt.show()