export_graphvizと私は「テキスト」に次のコードを使用していDT

Question

を可視化バイナリ分類問題：

def visualize_tree(tree,feature_names): 
    dot_data = StringIO() 
    export_graphviz(tree, 
        out_file=dot_data, 
        feature_names=feature_names, 
        special_characters=True)   
    graph = pydot.graph_from_dot_data(dot_data.getvalue(),) 
    graph.write_pdf("iris.pdf") 

vec = CountVectorizer(lowercase=True, tokenizer=tokens2, binary=True, ngram_range=(1,2)) 
x = vec.fit_transform(X_train) 
clf1 = DecisionTreeClassifier() 
clf1.fit(x, y_train)  
visualize_tree(clf1, vec.get_feature_names()) 

私はそれがこのような美しい木を生成しますfeature_names=feature_names,ずにを使用します。

私は木に余分な詳細を追加するfeature_names=feature_names,を追加するときしかし、それは私に、以下の「半木」を提供します！：

矢印なしで1行にすべて！どんな考え？私が試すことのできる代替手段はありますか？

Answer 1

よりもむしろpydotを使用して、あなたはちょうどあなたが空想になりたい場合は、subprocessとあなたのコードから呼び出すことができ、コマンドラインからgraphvizのを使用することができます。

import subprocess 

export_graphviz(model, 
       out_file='tree.dot', 
       feature_names=feature_names) 

subprocess.call(['dot', '-Tpdf', 'tree.dot', '-o' 'tree.pdf'])