2017-01-06 13 views
0

python scikit에서는 learn_graphviz라는 방법으로 의사 결정 트리를 도트 파일로 내보내는 방법이 있습니다.scikit-learn의 파일에서 결정 트리를로드하십시오.

의사 결정 트리로 도트 파일을 가져 오는 방법이 있는지 물어보고 싶습니다. 의사 결정 트리로 배우시겠습니까? sklearn.tree.import_graphviz()라는 일부 함수처럼?

+0

작성하기가 어려울 수 있습니다. 의사 결정 트리를 재 작성하는 데 필요한 정보는 "도트"파일로 옮겨지지 않고 손실됩니다. "도트"파일은 일반적인 목적의 유틸리티이며 결정 트리를 나타 내기 위해 특별히 설계되지 않은 graphviz를 지원하기위한 것입니다. 나는 문서에서 필요한 기능을 보지 못했다. – mikeTronix

+0

불가능합니다. 'graphviz'는 트리가 어떻게 그리고 어디서 나뉘어 졌는지 간단히 표현한 것입니다. –

답변

0

AFAIK 쉬운 방법은 없습니다. Graphviz은 의사 결정 트리를 시각화하는 데에만 사용할 수 있습니다. 모델을 저장하려면 Pickle을 사용하여 모델을 저장하십시오. 예 :

import cPickle 
# save the classifier 
with open('my_dumped_classifier.pkl', 'wb') as fid: 
    cPickle.dump(gnb, fid)  

# load it again 
with open('my_dumped_classifier.pkl', 'rb') as fid: 
    gnb_loaded = cPickle.load(fid)