Solucionado
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()
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.