Ainda não tem acesso? Estude com a gente! Matricule-se
Ainda não tem acesso? Estude com a gente! Matricule-se

Solucionado (ver solução)

Não exibe a arvore no pycharm

Não consigo exibir a árvore. Estou usando o pycharm. O código até gera o arquivo arvore, mas não exibe nada

dot_data = export_graphviz(modelo, out_file='arvore')
grafico = graphviz.Source(dot_data)
grafico

arquivo arvore digraph Tree { node [shape=box] ; 0 [label="X[2] <= 1.599\ngini = 0.04\nsamples = 1878097\nvalue = [1839885, 38212]"] ; 1 [label="X[4] <= 1.5\ngini = 0.483\nsamples = 38617\nvalue = [22901, 15716]"] ; 0 -> 1 [labeldistance=2.5, labelangle=45, headlabel="True"] ; 2 [label="gini = 0.0\nsamples = 17151\nvalue = [17151, 0]"] ; 1 -> 2 ; 3 [label="gini = 0.392\nsamples = 21466\nvalue = [5750, 15716]"] ; 1 -> 3 ; 4 [label="X[2] <= 3.67\ngini = 0.024\nsamples = 1839480\nvalue = [1816984, 22496]"] ; 0 -> 4 [labeldistance=2.5, labelangle=-45, headlabel="False"] ; 5 [label="gini = 0.19\nsamples = 94038\nvalue = [84023, 10015]"] ; 4 -> 5 ; 6 [label="gini = 0.014\nsamples = 1745442\nvalue = [1732961, 12481]"] ; 4 -> 6 ; }

2 respostas
solução

poderia adicionar o comando abaixo logo após "grafico = graphviz.Source(dot_data)" e executar novamente todos os passos.

grafico.render('test-output/modelo-tst.gv', view=True)

Obrigado Cleiton, após a sua sugestão a árvore foi exibida.