Solucionado (ver solução)
Solucionado
(ver solução)
1
resposta

Problemas para plotar a Arvore de Decisão

Estou com problemas para executar esse comando: salvar_arvore(classificador_arvore_decisao, "arvore_decisao1.png")

retorno:

NotFittedError: This DecisionTreeClassifier instance is not fitted yet. Call 'fit' with appropriate arguments before using this estimator.

Minhas funções:

def salvar_arvore (classificador, nome): plt.figure(figsize=(200,100)) tree.plot_tree(classificador, filled=True, fontsize=14) plt.savefig(nome) plt.clone()

classificador_arvore_decisao = tree.DecisionTreeClassifier()

1 resposta
solução!

Olá Lucas!

Na última linha do seu código, você está criando um objeto DecisionTreeClassifier()

classificador_arvore_decisao = tree.DecisionTreeClassifier()

Observe que após a criação do objeto, o modelo ainda não foi treinado, conforme foi notificado no erro NotFittedError: This DecisionTreeClassifier instance is not fitted yet

Você pode fazer isso usando o .fit(), substituindo X_train, y_train pelos seus dados de treino:

classificador_arvore_decisao.fit(X_train, y_train)

Após treinar o modelo, você poderá gerar a visualização da árvore de decisão. :D

Na documentação oficial do Sklearn você encontra alguns exemplos.