Decision Tree Visualizer

?? 의사 결정 트리 시각화

소개

의사 결정 트리 비주얼 라이저는 스카리 의사 결정 트리 분류기를 쉽게 시각화 할 수있는 강력한 라이브러리입니다. 트리 구조 및 규칙에 대한 유용한 정보를 추출하는 기능을 제공하고 의사 결정 트리를 시각화하기위한 HTML 파일을 생성합니다.

특징

• ? HTML 템플릿을 사용하여 Sklearn Decision 트리 분류기를 시각화합니다
• ? 트리 구조 및 규칙에 대한 유용한 정보를 추출하십시오
• 시각화를 위해 출력 HTML 파일을 생성합니다
• ? 더 나은 시각화를 위해 대상 이름과 색상을 사용자 정의하십시오

? 설치

라이브러리를 설치하려면 PIP를 사용하십시오.

 pip install d-treevis

용법

시작하려면 라이브러리를 가져 와서 create_tree 및 create_sankey 함수를 사용하십시오.

 import d_treevis as dtv

다음으로 데이터 세트에 Sklearn Decision Tree Classifier를 맞추고 create_tree 기능으로 전달하십시오.

 from sklearn . tree import DecisionTreeClassifier
    
# Fit a Decision Tree Classifier on your dataset
tree_model = DecisionTreeClassifier ()
tree_model . fit ( X , y )

# Visualize the decision tree
visualizer . visualize ( tree_model )

더 나은 시각화를 위해 대상 이름과 색상을 사용자 정의 할 수도 있습니다.

 # Define target names and colors
target_names = [ 'Survived' , 'Not Survived' ]
target_colors = [ 'red' , 'yellow' ]

# Visualize the decision tree with custom target names and colors
visualizer . visualize ( tree_model , target_names = target_names , target_colors = target_colors )

선적 서류 비치

자세한 문서 및 예는 의사 결정 트리 시각화 문서를 참조하십시오.

? 특허

이 라이브러리는 MIT 라이센스에 따라 라이센스가 부여됩니다.

? 기여

기부금을 환영합니다! 풀 요청을 제출하기 전에 기여 가이드 라인을 읽으십시오.

? 연락하다

질문이나 제안이 있으시면 [email protected]으로 문의하십시오.

? 참조

• Sklearn Decision Tree Classifier