sniffer_classifier.py 文件源码

def create_graphviz_file(self, file_name):
        dot_data = tree.export_graphviz(
            self.clf,
            out_file=None,
            feature_names=self.feature_names,
            class_names=self.target_names,
            filled=True, rounded=True,
            special_characters=True)
        graph = pdp.graph_from_dot_data(dot_data)
        graph.write_pdf(file_name + ".pdf")
        print "Decision graph created"