Olá Allan, tudo bem?
O erro indica que a variável tree
que não foi definida. Note que ao tentar usar plot_tree
, você está passando filled = tree
, mas deveria ser filled = True
.
O código corrigido é:
plt.figure(figsize = (15, 6))
plot_tree(arvore, filled = True, class_names = ['nao', 'sim'], fontsize = 7, feature_names = nome_colunas);
Espero ter ajudado.
Qualquer dúvida, não hesite em compartilhar no fórum.
Abraços!